POJ 1741 Tree(点分治)
2015-12-18 02:39
489 查看
题意:
N≤104的一棵树,(u,v,l),l≤103三元组描述这棵树,求dis(u,v)≤k,u≠v的点对数,k≤109
分析:
暴力的O(n2),以及dp的O(nk)肯定都是跑不过的,我们需要一个带log的算法,那么考虑分治
这里点分治,显然对于一棵树,为了不能退化每次应该选择一个特殊的点,删除这个点后得到的最大子树的顶点数应该最少
我们称这样删除这个点后得到的最大子树的顶点数最少的点叫做重心
事实上删除重心后,树的大小至少会减半
下面考虑证明存在这样的点具有重心的性质
选择任意顶点为起点,每次都向着最大子树的方向移动的话,最终定会到底一个点使得所有子树的大小不超过n/2
设当前顶点为u,如果u已经满足条件了那么算法应该终止,否则,某个与u相邻的顶点的子树的大小>n/2,假设它为点v
我们把v作为新的起点,不断重复这个过程,一定会在有限步终止
对于移动的边(u,v),u侧子树的大小<n/2,v测子树的大小>n/2,一定不会从v回到u,顶点数有限那么算法必然会在有限步终止
上面所说的也就是树形dp来求重心的过程
接着考虑如何统计点对,对于分治递归的过程中,我们可以顺便统计在同一颗子树的点对
但是最后归并的过程中,要减去它们,它们被重复统计了,挑战讲的非常棒
至此问题解决
代码:
N≤104的一棵树,(u,v,l),l≤103三元组描述这棵树,求dis(u,v)≤k,u≠v的点对数,k≤109
分析:
暴力的O(n2),以及dp的O(nk)肯定都是跑不过的,我们需要一个带log的算法,那么考虑分治
这里点分治,显然对于一棵树,为了不能退化每次应该选择一个特殊的点,删除这个点后得到的最大子树的顶点数应该最少
我们称这样删除这个点后得到的最大子树的顶点数最少的点叫做重心
事实上删除重心后,树的大小至少会减半
下面考虑证明存在这样的点具有重心的性质
选择任意顶点为起点,每次都向着最大子树的方向移动的话,最终定会到底一个点使得所有子树的大小不超过n/2
设当前顶点为u,如果u已经满足条件了那么算法应该终止,否则,某个与u相邻的顶点的子树的大小>n/2,假设它为点v
我们把v作为新的起点,不断重复这个过程,一定会在有限步终止
对于移动的边(u,v),u侧子树的大小<n/2,v测子树的大小>n/2,一定不会从v回到u,顶点数有限那么算法必然会在有限步终止
上面所说的也就是树形dp来求重心的过程
接着考虑如何统计点对,对于分治递归的过程中,我们可以顺便统计在同一颗子树的点对
但是最后归并的过程中,要减去它们,它们被重复统计了,挑战讲的非常棒
至此问题解决
代码:
// // Created by TaoSama on 2015-12-17 // Copyright (c) 2015 TaoSama. All rights reserved. // //#pragma comment(linker, "/STACK:1024000000,1024000000") #include <algorithm> #include <cctype> #include <cmath> #include <cstdio> #include <cstdlib> #include <cstring> #include <iomanip> #include <iostream> #include <map> #include <queue> #include <string> #include <set> #include <vector> using namespace std; #define pr(x) cout << #x << " = " << x << " " #define prln(x) cout << #x << " = " << x << endl const int N = 1e4 + 10, INF = 0x3f3f3f3f, MOD = 1e9 + 7; int n, k, ans; struct Edge { int v, nxt, c; } edge[N << 1]; int head , cnt; void addEdge(int u, int v, int c) { edge[cnt] = (Edge) {v, head[u], c}; head[u] = cnt++; } bool vis ; int sz , mx , centroid; int getAll(int u, int f) { int ret = 1; for(int i = head[u]; ~i; i = edge[i].nxt) { int v = edge[i].v; if(v == f || vis[v]) continue; ret += getAll(v, u); } return ret; } void getCentroid(int u, int f, int all) { sz[u] = 1; mx[u] = 0; for(int i = head[u]; ~i; i = edge[i].nxt) { int v = edge[i].v; if(v == f || vis[v]) continue; getCentroid(v, u, all); sz[u] += sz[v]; mx[u] = max(mx[u], sz[v]); } mx[u] = max(mx[u], all - sz[u]); if(!centroid || mx[u] < mx[centroid]) centroid = u; } void getDs(int u, int f, int dis, vector<int>& ds) { ds.push_back(dis); for(int i = head[u]; ~i; i = edge[i].nxt) { int v = edge[i].v; if(v == f || vis[v]) continue; getDs(v, u, dis + edge[i].c, ds); } } int calc(vector<int>& ds) { int ret = 0; sort(ds.begin(), ds.end()); for(int i = 0, j = ds.size() - 1; i < j; ++i) { while(i < j && ds[i] + ds[j] > k) --j; ret += j - i; } return ret; } void solve(int u) { centroid = 0; getCentroid(u, -1, getAll(u, -1)); int s = centroid; vis[s] = true; for(int i = head[s]; ~i; i = edge[i].nxt) { int v = edge[i].v; if(vis[v]) continue; solve(v); } vector<int> ds; ds.push_back(0); for(int i = head[s]; ~i; i = edge[i].nxt) { int v = edge[i].v; if(vis[v]) continue; vector<int> tds; getDs(v, s, edge[i].c, tds); ans -= calc(tds); ds.insert(ds.end(), tds.begin(), tds.end()); } ans += calc(ds); vis[s] = false; } int main() { #ifdef LOCAL freopen("C:\\Users\\TaoSama\\Desktop\\in.txt", "r", stdin); // freopen("C:\\Users\\TaoSama\\Desktop\\out.txt","w",stdout); #endif ios_base::sync_with_stdio(0); while(scanf("%d%d", &n, &k) == 2 && (n || k)) { cnt = 0; memset(head, -1, sizeof head); for(int i = 1; i < n; ++i) { int u, v, c; scanf("%d%d%d", &u, &v, &c); addEdge(u, v, c); addEdge(v, u, c); } ans = 0; solve(1); printf("%d\n", ans); } return 0; }
相关文章推荐
- bzoj-1095 Hide 捉迷藏
- hdu-5571 tree
- 【BZOJ2051】A Problem For Fun
- spoj1825 Free tour II 点分治
- POJ1741Tree(树上点分治)
- hdu5593 树形dp
- POJ 1987 Distance Statistics (树上点分治)
- POJ 1741 Tree (树分治之点分治)
- HDU 4812 D Tree (树分治之点分治)
- HDU 4871 Shortest-path tree (树分治之点分治 最短路径树问题)
- poj 2114 Boatherds 点分治
- 点分治专题——bzoj 1468 &bzoj 2152 题解
- POJ 1741 Tree + POJ 1987 Distance Statistics【树的点分治】
- BZOJ 1468: Tree
- POJ 2114 Boatherds
- poj 1741 点分治论文题
- poj 2114 点分治
- BZOJ 2152: 聪聪可可 点分治/树dp
- [bzoj 1758] [Wc2010]重建计划
- [BZOJ3697]采药人的路径(点分治)