題意：求一棵樹上任意兩點之間距離小于k的點對有多少個

分析：樹分治的練手題，找到重心以后，我們要求所有經過中心的點，如何求這樣的點呢，顯然，所有的路徑可以分為過重心，不過中心，終點是重心，所有點到達重心的距離很好求，可以很容易的求出，這些邊由兩部分組成，來自不同的兒子和來自相同的兒子，只要減去來自相同的兒子的數量就是過重心的數量

#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<vector> using namespace std; const int maxn=2e4+5; typedef long long ll; typedef unsigned long long ull; int vis[maxn],n,k,num,d[maxn],_min,tsize[maxn],tssize[maxn],root; int head[maxn],next[maxn],edge[maxn],val[maxn],tot; ll ans; void init(){memset(vis,0,sizeof(vis));tot=ans=0;memset(head,-1,sizeof(head)); }void addedge(int u,int v,int w){edge[tot]=v;val[tot]=w;next[tot]=head[u];head[u]=tot++; } //獲取子樹大小和子樹的最大值 void getsize(int u,int f){tsize[u]=1;tssize[u]=0;for(int i=head[u];i!=-1;i=next[i]){int v=edge[i];if(v==f||vis[v])continue;getsize(v,u);tsize[u]+=tsize[v];tssize[u]=max(tssize[u],tsize[v]);} } //重心 void getroot(int r,int u,int f){tssize[u]=max(tssize[u],tsize[r]-tsize[u]);if(tssize[u]<_min)_min=tssize[u],root=u;for(int i=head[u];i!=-1;i=next[i]){int v=edge[i];if(v==f||vis[v])continue;getroot(r,v,u);} } //求所有點到u的距離 void getdis(int u,int dcnt,int f){d[num++]=dcnt;for(int i=head[u];i!=-1;i=next[i]){int v=edge[i];int w=val[i];if(v==f||vis[v])continue;getdis(v,dcnt+w,u);} }int cal(int u,int dcnt){num=0;int ret=0;getdis(u,dcnt,-1);int i=0,j=num-1;sort(d,d+num);//for(int i=0;i<num;i++)cout<<d[i]<<" ";cout<<endl;while(i<j){while(d[i]+d[j]>k&&i<j)j--;ret+=j-i;i++;}//cout<<ret<<endl;return ret; }void dfs(int u){_min=n;getsize(u,-1);getroot(u,u,-1);//cout<<root<<endl;ans+=cal(root,0);//cout<<root<<" "<<ans<<endl;vis[root]=1;int qqq=root;for(int i=head[qqq];i!=-1;i=next[i]){int v=edge[i],w=val[i];if(vis[v])continue;ans-=cal(v,w);dfs(v);}//cout<<root<<" "<<ans<<endl; }int main(){//freopen("in","r",stdin);ios::sync_with_stdio(false);while(cin>>n>>k&&n+k){init();for(int i=0;i<n-1;i++){int u,v,w;cin>>u>>v>>w;addedge(u,v,w);addedge(v,u,w);}dfs(1);cout<<ans<<endl;}return 0; }

?

轉載于:https://www.cnblogs.com/jihe/p/5954284.html

總結

以上是生活随笔為你收集整理的poj 1741的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。