标签：
去网上搜题解大多数都是说论文，搜了论文看了看，写的确实挺好，直接复制过来。
不过代码中有些细节还是要注意的，参考这篇http://blog.sina.com.cn/s/blog_6d5aa19a0100o73m.html一段
设X为满足i<j且Depth[i]+Depth[j]<=K的数对(i,j)的个数
设Y为满足i<j，Depth[i]+Depth[j]<=K且Belong[i]=Belong[j]数对(i,j)的个数
那么我们要统计的量便等于X-Y
求X、Y的过程均可以转化为以下问题：
已知A[1],A[2],...A[m]，求满足i<j且A[i]+A[j]<=K的数对(i,j)的个数
对于这个问题，我们先将A从小到大排序。
设B[i]表示满足A[i]+A[p]<=K的最大的p(若不存在则为0)。我们的任务便转化为求出A所对应的B数组。那么，若B[i]>i，那么i对答案的贡献为B[i]-i。
显然，随着i的增大，B[i]的值是不会增大的。利用这个性质，我们可以在线性的时间内求出B数组，从而得到答案。
不过我的代码中这个算的还是有点不一样，他是算i的贡献的时候用B[i] - i, 他是从i开始往后数的，我是从他结束的那个店B[i]往前数的，一直数到最前头，所以代码也有所差别，具体细节见代码，代码中有注释。
#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int maxn = 22000; const int inf = 0x3f3f3f3f; int head[maxn], tot; struct Edge { int to, next, w; }edge[maxn]; bool vis[maxn]; int dep[maxn]; int siz[maxn]; int le, ri; int minn; int n, k; void init() { tot = 0; memset(head, -1, sizeof(head)); memset(vis, false, sizeof(vis)); } void addedge(int u, int v, int w) { edge[tot].to = v; edge[tot].w = w; edge[tot].next = head[u]; head[u] = tot++; } int dfs_size(int u, int fa)//dfs找出每个节点对应的子树的大小 { siz[u] = 1; for (int i = head[u]; i != -1; i = edge[i].next) { int v = edge[i].to; if (v == fa || vis[v]) continue; siz[u] += dfs_size(v, u); } return siz[u]; } void dfs_depth(int u, int fa, int d)//每个节点到根节点的距离 { dep[ri++] = d; for (int i = head[u]; i != -1; i = edge[i].next) { int v = edge[i].to; if (v == fa || vis[v]) continue; dfs_depth(v, u, d + edge[i].w); } } void get_focus(int u, int fa, int totnum, int &root)//找出树的重心 { int m = totnum - siz[u]; for (int i = head[u]; i != -1; i = edge[i].next) { int v = edge[i].to; if (v == fa || vis[v]) continue; get_focus(v, u, totnum, root); m = max(m, siz[v]);//找出最大子树 } if (m < minn)//最大子树的最小的数值即重心对应的位置 { minn = m; root = u; } } int calc(int a, int b)//计算一共有多少符合条件的点对 { int e = b - 1, ans = 0; sort(dep + a, dep + b); for (int i = a; i < b; i++)//对i来说，它的贡献是多大，枚举每一个i { if (dep[i] > k) break;//如果他大于k了，就说明肯定不满足 while (e >= a && dep[e] + dep[i] > k) e--;//随着i的增大，e肯定是递减的，只要找到i对应的最大的e小于等于k，那么这个i的贡献就是e-a+1,就是e之前的所有的 ans += e - a + 1; if (e > i) ans--;//这里是在算他的贡献的时候，如果e大于i的话，那么加到答案里肯定有个i，所以要减去1 } return ans >> 1;//一对点算了两次，要除以2 } int solve(int u) { int totnum = dfs_size(u, 0); int ans = 0; minn = inf; int root; get_focus(u, 0, totnum, root);//找到以u为根节点子树的重心root vis[root] = true; for (int i = head[root]; i != -1; i = edge[i].next) { int v = edge[i].to; if (vis[v]) continue; ans += solve(v);//把所有子树的个数加进来 } le = ri = 0; for (int i = head[root]; i != -1; i = edge[i].next) { int v = edge[i].to; if (vis[v]) continue; dfs_depth(v, root, edge[i].w);//找出v的子树里面的 ans -= calc(le, ri); le = ri; } ans += calc(0, ri);//这里计算就是论文里说的式子：总共的-子树当中的 for (int i = 0; i < ri; i++) { if (dep[i] <= k) ans++; else break; } vis[root] = false; return ans; } int main() { while (~scanf("%d%d", &n, &k)) { if (n == 0 && k == 0) break; init(); int u, v, w; for (int i = 1; i < n; i++) { scanf("%d%d%d", &u, &v, &w); addedge(u, v, w); addedge(v, u, w); } printf("%d\n", solve(1)); } return 0; }
标签：
原文地址：http://www.cnblogs.com/Howe-Young/p/4776852.html
