VK Cup 2012 Finals E

来源：互联网发布：驼峰命名 c语言编辑：程序博客网时间：2024/04/24 11:36

http://codeforces.com/contest/212/problem/E

才做了一道题，就调试了我一个下午的时间，太弱了快哭了；

题意：就是给定最多n个点的树，要求除去某个点后，在剩下的至少两个联通分量中，分配到两个集合，着两个集合就是对应题目给定的两个餐厅

思路：枚举以每个点为根，然后dp下就可以了

#include <iostream>#include <cstdio>#include <vector>#include <algorithm>#include <cstring>using namespace std;const int maxn=5010;vector<int> g[maxn];int dp[maxn],n;bool mark[maxn][maxn],ans[maxn];int s[maxn];void dfs(int u,int fa){    mark[u][0]=1;    for(int i=0;i<g[u].size();i++)    {        if(g[u][i]==fa) continue;        int v=g[u][i];        dfs(v,u);        for(int j=n-1;j>=0;j--)           if(mark[u][j]) mark[u][j+dp[v]+1]=1;        dp[u]+=dp[v]+1;    }     for(int j=n-1;j>=0;j--)       if(mark[u][j]&&j+n-1-dp[u]<n) mark[u][j+n-1-dp[u]]=1;     for(int j=n-1;j>=0;j--)       if(mark[u][j]) ans[j]=1;}int main(){    int a,b;    while(scanf("%d",&n)==1)    {        for(int i=1;i<n;i++)        {            scanf("%d%d",&a,&b);            g[a].push_back(b);            g[b].push_back(a);        }        memset(dp,0,sizeof(dp));        memset(ans,0,sizeof(ans));        dfs(1,-1);       /* for(int i=1;i<=n;i++)          cout<<ans[i]<<" ";        cout<<endl;*/        int k=0;        for(int i=1;i<n-1;i++)         if(ans[i]) s[k++]=i;        printf("%d\n",k);        for(int i=0;i<k;i++)          printf("%d %d\n",s[i],n-1-s[i]);        for(int i=1;i<=n;i++)  g[i].clear();    }    return 0;}