博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
POJ 1741 Tree ——点分治
阅读量:6994 次
发布时间:2019-06-27

本文共 1573 字,大约阅读时间需要 5 分钟。

【题目分析】

    这貌似是做过第三道以Tree命名的题目了。

    听说树分治的代码都很长,一直吓得不敢写,有生之年终于切掉这题。

    点分治模板题目。自己YY了好久才写出来。

    然后1A了,开心o(* ̄▽ ̄*)ブ

【代码】

#include 
#include
#include
#include
#define maxn 20005#define inf 0x3f3f3f3fusing namespace std;int n,k,h[maxn],to[maxn],ne[maxn],w[maxn],en,cnt;int a[maxn],b[maxn],ban[maxn],siz[maxn],size,root;int mx[maxn],now,now_mx,tot;void init(){en=cnt=0;memset(h,-1,sizeof h);memset(ban,0,sizeof ban);}void add(int a,int b,int c){to[en]=b;w[en]=c;ne[en]=h[a];h[a]=en++;}void rd(){for(int i=1;i
=0;i=ne[i]) if (to[i]!=fa&&!ban[to[i]]){ dfs_size(to[i],o); siz[o]+=siz[to[i]]; mx[o]=max(mx[o],siz[to[i]]); }}void dfs_root(int o,int fa){ int now=max(mx[o],size-siz[o]); if (now
=0;i=ne[i]) if (to[i]!=fa&&!ban[to[i]]) dfs_root(to[i],o);}void dfs_dist(int o,int fa,int dist){ a[++tot]=dist; for (int i=h[o];i>=0;i=ne[i]) if (to[i]!=fa&&!ban[to[i]]) dfs_dist(to[i],o,dist+w[i]);}int cal(){ int ret=0,j=tot; sort(a+1,a+tot+1); for (int i=1;i<=tot;++i) { while (a[j]+a[i]>k&&j) j--; ret+=j; if (j>i) ret--; } return ret/2;}void dfs(int o){ dfs_size(o,0); size=siz[o]; now_mx=inf; dfs_root(o,0); ban[root]=1; for (int i=h[root];i>=0;i=ne[i]) { if (!ban[to[i]]) { tot=0; dfs_dist(to[i],root,w[i]); cnt-=cal(); } } tot=0; dfs_dist(root,0,0); cnt+=cal(); for (int i=h[root];i>=0;i=ne[i]) if (!ban[to[i]]) dfs(to[i]);}int main(){ while (scanf("%d%d",&n,&k)!=EOF&&n&&k) { init();rd();dfs(1); printf("%d\n",cnt); }}

  

转载于:https://www.cnblogs.com/SfailSth/p/6360941.html

你可能感兴趣的文章