2017年多校赛第三场 1005 RXD and dividing（DFS）

当时写这道题目的时候实在是懵了。研究了好久什么叫做斯塔纳树。就觉得很混乱。

我现在对斯塔纳树的理解就是，选取若干个点，然后生成一棵数即可。两点之间的边肯定是走最短路的。但是题目说的最小斯塔纳树又是啥呢。。。

后来看了看题解，也没说啥叫斯塔纳树，但是就是直接DFS求贡献度即可。

代码如下：

#include<iostream>#include<cstdio>#include<vector>#include<queue>#include<utility>#include<stack>#include<algorithm>#include<cstring>#include<string>using namespace std;const int maxn = 1e6 + 5;int n, k;int head[maxn], to[maxn << 1], nx[maxn << 1], ppp;long long cost[maxn << 1];long long ans;inline void add_edge(int u, int v, long long c) {to[ppp] = v, nx[ppp] = head[u], cost[ppp] = c, head[u] = ppp++;to[ppp] = u, nx[ppp] = head[v], cost[ppp] = c, head[v] = ppp++;}int dfs(int last, int u, int id) {int cnt = 1;for(int i = head[u]; ~i; i = nx[i]) {if(to[i] == last)continue;cnt += dfs(u, to[i], i);}ans += min(cnt, k) * cost[id];return cnt;}int main() {while(scanf("%d%d", &n, &k) != EOF) {ppp = 0;memset(head, -1, sizeof(head));ans = 0;int a, b;long long c;for(int i = 1; i < n; i++) {scanf("%d%d%I64d", &a, &b, &c);add_edge(a, b, c);}for(int i = head[1]; ~i; i = nx[i]) {dfs(1, to[i], i);}printf("%I64d\n", ans);} return 0;}