【BZOJ】【P1468】【Tree】【题解】【点分治】

来源：互联网发布：查股网ddx日线数据编辑：程序博客网时间：2024/06/05 05:42

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=1468

刚学点分治,简要的记录一下

每次选择一个点把这棵树分成几个不相交的子树

挑一个使得最大子树size最小的点

计算过这个点的答案

递归处理这几个子树

Code:

#include<cstdio>#include<iostream>#include<algorithm>#include<vector>using namespace std;const int maxn=40005;struct edge{int u,v,w;};vector<edge>G[maxn];int n,m,k,root,All,ans=0;int siz[maxn],f[maxn],d[maxn],data[maxn];bool cant[maxn];void makert(int u,int fa){siz[u]=1;f[u]=0;for(int i=0;i<G[u].size();i++){edge e=G[u][i];if(fa==e.v||cant[e.v])continue;makert(e.v,u);siz[u]+=siz[e.v];f[u]=max(f[u],siz[e.v]);}f[u]=max(f[u],All-siz[u]);if(f[root]>f[u])root=u;}void dfs(int u,int fa){data[++data[0]]=d[u];for(int i=0;i<G[u].size();i++){edge e=G[u][i];if(e.v!=fa&&!cant[e.v]){d[e.v]=d[u]+e.w;dfs(e.v,u);}}}int calc(int u){data[0]=0;dfs(u,0);int ans=0,l=1,r=data[0];sort(data+1,data+1+data[0]);while(l<r){if(data[l]+data[r]<=k)ans+=(r-l),l++;else r--;}return ans;}void solve(int u){d[u]=0;ans+=calc(u);cant[u]=1;for(int i=0;i<G[u].size();i++){edge e=G[u][i];if(cant[e.v])continue;d[e.v]=e.w;ans-=calc(e.v);f[root=0]=n+1;All=siz[e.v];makert(e.v,0);solve(root);}}int main(){scanf("%d",&n);All=n;for(int i=1;i<n;i++){int u,v,w;scanf("%d%d%d",&u,&v,&w);G[u].push_back((edge){u,v,w});G[v].push_back((edge){v,u,w});}scanf("%d",&k);f[root]=n+1;makert(1,0);solve(root);cout<<ans<<endl;return 0;}

0 0