POJ 2378 Tree Cutting 3140 Contestants Division (简单树形dp)

POJ 2378 Tree Cutting:题意求删除哪些单点后产生的森林中的每一棵树的大小都小于等于原树大小的一半

 #include<cstdio>

 #include<cstring>

 #include<iostream>

 #include<algorithm>

 #include<cmath>

 using namespace std;

 const int maxn=;

 const int maxx=<<;

 int t,n;

 int f[maxn]={};

 int vis[maxn]={};

 struct nod{

     int y;

     int next;

 }e[*maxn]={};

 int head[maxn]={},tot=;

 int ans[maxn]={},size=,sum=;

 void init(int x,int y){

     e[++tot].y=y;

     e[tot].next=head[x];

     head[x]=tot;

 }

 void dfs(int x){

     int y;

     vis[x]=;

     int tmp=;

     for(int i=head[x];i;i=e[i].next){

         y=e[i].y;

         if(!vis[y]){

             dfs(y);

             f[x]+=f[y]+;

             tmp=max(f[y]+,tmp);

         }

     }

     tmp=max(tmp,n-f[x]-);

     if(tmp<size){

         ans[++sum]=x;

     }

 }

 int main(){

     scanf("%d",&n);

     size=n/+;

     int x,y;

     for(int i=;i<n;i++){

         scanf("%d%d",&x,&y);

         init(x,y);

         init(y,x);

     }

     dfs();

     sort(ans+,ans++sum);

     for(int i=;i<=sum;i++){

         printf("%d\n",ans[i]);

     }

     return ;

 }

POJ 3140 Contestants Division:题意删除某边后剩余两个联通块节点数的差最小,输出这个差,

要用longlong,记得把上限设大...

 #include<cstdio>

 #include<cstring>

 #include<iostream>

 #include<algorithm>

 #include<cmath>

 using namespace std;

 const int maxn=;

 long long maxx=2000000000000000000LL;

 int t,n,m;

 long long f[maxn]={},v[maxn]={};

 int vis[maxn]={};

 struct nod{

     int y;

     int next;

 }e[*maxn]={};

 int head[maxn]={},tot;

 long long ans,sum,bign;

 void init(int x,int y){

     e[++tot].y=y;

     e[tot].next=head[x];

     head[x]=tot;

 }

 inline long long mabs(long long x){

     if(x<){

         return -x;

     }

     return x;

 }

 void dfs(int x){

     int y;

     long long tmp=;

     vis[x]=;

     for(int i=head[x];i;i=e[i].next){

         y=e[i].y;

         if(!vis[y]){

             dfs(y);

             f[x]+=f[y];

             tmp=mabs(bign-f[y]-f[y]);

             ans=min(ans,tmp);

         }

     }

     f[x]+=v[x];

 }

 void yu(){

     bign=sum=;

     tot=;

     ans=maxx;

     memset(f,,sizeof(f));

     memset(vis,,sizeof(vis));

     memset(head,,sizeof(head));

     memset(v,,sizeof(v));

 }

 int main(){

     for(int zz=;;zz++){

         yu();

         scanf("%d%d",&n,&m);

         if(n==&&m==){

             break;

         }

         int x,y;

         for(int i=;i<=n;i++){

             scanf("%lld",&v[i]);

             bign+=v[i];

         }

         for(int i=;i<=m;i++){

             scanf("%d%d",&x,&y);

             init(x,y);

             init(y,x);

         }

         dfs();

         printf("Case %d: %lld\n",zz,ans);

     }

     return ;

 }

巴特西

POJ 2378 Tree Cutting 3140 Contestants Division (简单树形dp)

最新文章

热门文章