POJ1741&&BZOJ1468 男人八题

来源：互联网发布：臭氧消毒知乎编辑：程序博客网时间：2024/06/05 08:40

去年这个时候就做这道题了那个时候完全不懂

今天重新来学了一下点分治发现自己很多不会

没有照着别人的程序打点分治的时候思路很乱可能是自己太弱了

后来看了黄学长的代码发现有些东西我打的太复杂了。。。。

然后我就像到了一个新世界

每次都去找中心就好了。。。。

#include<cstdio>#include<iostream>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;char c;bool flag;inline void read(int &a){a=0,flag=false;do c=getchar();while(c!='-'&&(c<'0'||c>'9'));if(c=='-')flag=true,c=getchar();while(c<='9'&&c>='0')a=(a<<3)+(a<<1)+c-'0',c=getchar();   if(flag)a=-a;}int root;struct Chain{Chain *next;int u,len;Chain(){next=NULL;}}*Head[100001];bool choose[10001];int sum;int Max[100001],son[100001];void Get_Root(int u,int fa){son[u]=1;Max[u]=0;for(Chain *tp=Head[u];tp;tp=tp->next){if(tp->u==fa||choose[tp->u])continue;Get_Root(tp->u,u);son[u]+=son[tp->u];Max[u]=max(Max[u],son[tp->u]);}Max[u]=max(Max[u],sum-son[u]);if(Max[u]<Max[root])root=u;}int deep[100001];inline void addside(int a,int b,int len){Chain *tp=new Chain;tp->u=b,tp->len=len,tp->next=Head[a];Head[a]=tp;swap(a,b);    tp=new Chain;tp->u=b,tp->len=len,tp->next=Head[a];Head[a]=tp;}int D[100001];inline void De(int u,int fa){for(Chain *tp=Head[u];tp;tp=tp->next)        if(tp->u!=fa&&!choose[tp->u])           D[tp->u]=D[u]+tp->len,deep[++deep[0]]=D[tp->u],De(tp->u,u);}int k;int Sol_dep(int u,int delta){int res=0;       deep[0]=0;   D[u]=deep[++deep[0]]=delta;   De(u,u);       sort(deep+1,deep[0]+deep+1);       int l=1,r=deep[0];       while(l<r)       {            if(deep[l]+deep[r]>k)     r--;      else res+=r-l,l++;   }   return res;}int ans;int Div(int u){ans+=Sol_dep(u,0);choose[u]=true;for(Chain *tp=Head[u];tp;tp=tp->next)  {    if(choose[tp->u])continue;    ans-=Sol_dep(tp->u,tp->len);    sum=son[tp->u];    root=0;    Get_Root(tp->u,root);    Div(root);  }}int main(){ int a,b,c;while(true){int n;read(n),read(k);memset(choose,false,sizeof(choose));if(!n)return 0;for(int i=1;i<=n;i++)  Head[i]=NULL;for(int i=1;i<n;i++)   {   read(a),read(b),read(c),addside(a,b,c);   }sum=n;root=0;Max[0]=1<<29;ans=0;Get_Root(1,0);Div(root);printf("%d\n",ans);}return 0;}

0 0