NC13886 Shortest Path
Shortest Path
https://ac.nowcoder.com/acm/problem/13886
NC13886 Shortest Path
题目地址:
基本思路:
前几天好像就做了类似的题,当时既要求最小也要求最大,基本思路都是贪心+dfs;
这里我们要求最小值:
所以每一个节点尽可能在子树内部跟自己相邻的节点配对解决。对于每一条边,如果是sz[v]是奇数,则被计算1次,否则不需要被计算,因为是偶数完全可以在内部配对,所以一遍dfs并且记录答案就好了。
拓展
我们可以思考如果要求最大值怎么办:
这时我们让每一个节点尽可能经过根节点,跟根节点的另外一个子树配对。我们同样计录每一条边被计算的次数。假设这条边连接的是u,v两个节点,u是父亲节点,v是儿子节点,则该边被计算的次数为min(sz[v],n-sz[v])
。
参考代码:
#pragma GCC optimize(2) #pragma GCC optimize(3) #include <bits/stdc++.h> using namespace std; #define IO std::ios::sync_with_stdio(false) #define int long long #define rep(i, l, r) for (int i = l; i <= r; i++) #define per(i, l, r) for (int i = l; i >= r; i--) #define mset(s, _) memset(s, _, sizeof(s)) #define pb push_back #define pii pair <int, int> #define mp(a, b) make_pair(a, b) #define INF 1e18 inline int read() { int x = 0, neg = 1; char op = getchar(); while (!isdigit(op)) { if (op == '-') neg = -1; op = getchar(); } while (isdigit(op)) { x = 10 * x + op - '0'; op = getchar(); } return neg * x; } inline void print(int x) { if (x < 0) { putchar('-'); x = -x; } if (x >= 10) print(x / 10); putchar(x % 10 + '0'); } const int maxn = 1e5 + 10; struct Edge{ int to,next,val; }edge[2*maxn]; int n,ans; int cnt = 0,head[maxn]; void add_edge(int u,int v,int w){ edge[++cnt].next = head[u]; edge[cnt].to = v; edge[cnt].val = w; head[u] = cnt; } int sz[maxn]; void dfs(int u,int p) { sz[u] = 1; for (int i = head[u]; i != -1; i = edge[i].next) { int to = edge[i].to; if (to == p) continue; dfs(to, u); sz[u] += sz[to]; if (sz[to] % 2) ans += edge[i].val; } } signed main() { IO; int t; cin >> t; while (t-->0) { cin >> n; cnt = 0; mset(head, -1); mset(sz,0); ans = 0; rep(i, 1, n-1) { int u, v, w; cin >> u >> v >> w; add_edge(u, v, w); add_edge(v, u, w); } dfs(1, 0); cout << ans << endl; } return 0; }