树形dp练习

/*poj 1463 最小点覆盖 匈牙利*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 1510
using namespace std;
int n,num,head[maxn],match[maxn],c[maxn],ans;
bool f[maxn];
struct node{
    int v,pre;
}e[maxn*];
int init(){
    int x=;char s=getchar();
    while(s<''||s>'')s=getchar();
    while(s>=''&&s<=''){x=x*+s-'';s=getchar();}
    return x;
}
void Clear(){
    memset(c,,sizeof(c));
    memset(head,,sizeof(head));
    memset(match,,sizeof(match));
    num=ans=;
}
void Add(int from,int to){
    num++;e[num].v=to;
    e[num].pre=head[from];
    head[from]=num;
}
void Dfs(int now,int from,int dep){
    c[now]=dep;
    for(int i=head[now];i;i=e[i].pre){
        int v=e[i].v;
        if(v==from)continue;
        Dfs(v,now,dep+);
    }
}
int dfs(int u){
    for(int i=head[u];i;i=e[i].pre){
        int v=e[i].v;
        if(f[v])continue;f[v]=;
        if(match[v]==||dfs(match[v])){
            match[v]=u;return ;
        }
    }
    return ;
}
int main()
{
    while(scanf("%d",&n)!=EOF){
        int u,v,s;Clear();
        for(int i=;i<=n;i++){
            u=init();s=init();
            while(s--){
                v=init();Add(u,v);Add(v,u);
            }
        }
        Dfs(,-,);
        for(int i=;i<n;i++)if(c[i]&){
            memset(f,,sizeof(f));
            ans+=dfs(i);
        }
        printf("%d\n",ans);
    }
    return ;
}

/*poj 1463 最小点覆盖 dp*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 1510
using namespace std;
int n,num,head[maxn],f[maxn][];
struct node{
    int v,pre;
}e[maxn*];
int init(){
    int x=;char s=getchar();
    while(s<''||s>'')s=getchar();
    while(s>=''&&s<=''){x=x*+s-'';s=getchar();}
    return x;
}
void Add(int from,int to){
    num++;e[num].v=to;
    e[num].pre=head[from];
    head[from]=num;
}
void Dfs(int x,int from){
    int s=;f[x][]=;
    for(int i=head[x];i;i=e[i].pre){
        int v=e[i].v;
        if(v==from)continue;
        Dfs(v,x);s++;
        f[x][]+=min(f[v][],f[v][]);
        f[x][]+=f[v][];
    }
}
int main()
{
    while(scanf("%d",&n)!=EOF){
        int u,v,s;num=;
        memset(f,,sizeof(f));
        memset(head,,sizeof(head));
        for(int i=;i<=n;i++){
            u=init();s=init();
            while(s--){
                v=init();Add(u,v);Add(v,u);
            }
        }
        Dfs(,-);
        printf("%d\n",min(f[][],f[][]));
    }
    return ;
}

/*f[i][j][0/1] i子树走j步是否回到i的max*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 210
using namespace std;
int n,m,num,head[maxn],f[maxn][maxn][],c[maxn],ans;
struct node{
    int v,pre;
}e[maxn*];
void Add(int from,int to){
    num++;e[num].v=to;
    e[num].pre=head[from];
    head[from]=num;
}
void Clear(){
    num=;ans=;
    memset(f,,sizeof(f));
    memset(head,,sizeof(head));
}
void Dfs(int x,int from){
    f[x][][]=c[x];
    for(int i=head[x];i;i=e[i].pre){
        int v=e[i].v;
        if(v==from)continue;
        Dfs(v,x);
        for(int j=m;j>=;j--)
            for(int k=;k<=j;k++){
                if(j-k->=)f[x][j][]=max(f[x][j][],f[x][j-k-][]+max(f[v][k][],f[v][k][]));
                if(j-k->=)f[x][j][]=max(f[x][j][],f[x][j-k-][]+f[v][k][]);
                //这里一开始漏掉了 可以先走v 也可以最后走v
                if(j-k->=)f[x][j][]=max(f[x][j][],f[x][j-k-][]+f[v][k][]);
            }
    }
}
int main()
{
    while(~scanf("%d%d",&n,&m)){
        Clear();int u,v;
        for(int i=;i<=n;i++)
            scanf("%d",&c[i]);
        for(int i=;i<n;i++){
            scanf("%d%d",&u,&v);
            Add(u,v);Add(v,u);
        }
        Dfs(,-);
        for(int i=;i<=m;i++){
            ans=max(ans,f[][i][]);
            ans=max(ans,f[][i][]);
        }
        printf("%d\n",ans);
    }
    return ;
}

/*poj 1947*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 210
#define inf 0x3f3f3f3f
using namespace std;
int n,p,num,head[maxn],root,f[maxn][maxn],l[maxn],r[maxn],fa[maxn],son[maxn][maxn],sum[maxn];
struct node{
    int v,pre;
}e[maxn];
void Add(int from,int to){
    num++;e[num].v=to;
    e[num].pre=head[from];
    head[from]=num;
}
void Dfs(int now,int from){
    fa[now]=from;
    for(int i=head[now];i;i=e[i].pre){
        int v=e[i].v;
        if(v==from)continue;
        son[now][++son[now][]]=v;
        Dfs(v,now);
    }
}
int DP(int now){
    if(!now)return ;
    int s=;
    for(int i=;i<=son[now][];i++)
        s+=DP(son[now][i]);
    sum[now]=s+;
    f[now][]=son[now][];
    for(int i=;i<=son[now][];i++){
        int x=son[now][i];
        for(int j=sum[now];j>=;j--)
            for(int k=;k<=sum[x]&&k<=j;k++)
                f[now][j]=min(f[now][j],f[now][j-k]+f[x][k]-);
    }
    return sum[now];
}
int main()
{
    scanf("%d%d",&n,&p);
    int u,v;
    for(int i=;i<n;i++){
        scanf("%d%d",&u,&v);
        Add(u,v);Add(v,u);
    }
    Dfs(,);memset(f,/,sizeof(f));
    sum[]=DP();int ans=f[][p];
    for(int i=;i<=n;i++)
        ans=min(ans,f[i][p]+);
    printf("%d\n",ans);
    return ;
}

/*poj 3342 树上最大独立集+方案数统计*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
#define maxn 210
using namespace std;
int n,num,head[maxn],f[maxn][],g[maxn][],cnt;
map<string,int>r;
struct node{
    int v,pre;
}e[maxn*];
void Add(int from,int to){
    num++;e[num].v=to;
    e[num].pre=head[from];
    head[from]=num;
}
void Dfs(int x,int from){
    g[x][]=;f[x][]=;g[x][]=;
    int sum,mx;
    for(int i=head[x];i;i=e[i].pre){
        int v=e[i].v;
        if(v==from)continue;Dfs(v,x);
        mx=max(f[v][],f[v][]);sum=;
        if(mx==f[v][])sum+=g[v][];
        if(mx==f[v][])sum+=g[v][];
        f[x][]+=mx;g[x][]*=sum;
        mx=f[v][];sum=g[v][];
        f[x][]+=mx;g[x][]*=sum;
    }
}
int main()
{
    while(){
        cin>>n;if(n==)break;
        cnt=;num=;r.clear();
        memset(f,,sizeof(f));
        memset(g,,sizeof(g));
        memset(head,,sizeof(head));
        string u,v,root;
        cin>>root;r[root]=++cnt;
        for(int i=;i<n;i++){
            cin>>u>>v;
            if(r[u]==)r[u]=++cnt;
            if(r[v]==)r[v]=++cnt;
            Add(r[u],r[v]);
            Add(r[v],r[u]);
        }
        Dfs(r[root],);
        int mx,sum=;
        mx=max(f[r[root]][],f[r[root]][]);
        if(mx==f[r[root]][])sum+=g[r[root]][];
        if(mx==f[r[root]][])sum+=g[r[root]][];
        printf("%d ",mx);
        if(sum==)printf("Yes\n");
        else printf("No\n");
    }
}

/*codevs 1378 选课 背包搞*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 310
using namespace std;
int n,m,f[maxn][maxn],son[maxn][maxn],num,head[maxn],v[maxn],c[maxn];
struct node{
    int v,pre;
}e[maxn*];
int init(){
    int x=,f=;char s=getchar();
    while(s<''||s>''){if(s=='-')f=-;s=getchar();}
    while(s>=''&&s<=''){x=x*+s-'';s=getchar();}
    return x*f;
}
void Add(int from,int to){
    num++;e[num].v=to;
    e[num].pre=head[from];
    head[from]=num;
}
void Get(int now,int dep){
    c[now]=dep;
    for(int i=head[now];i;i=e[i].pre){
        son[now][++son[now][]]=e[i].v;
        Get(e[i].v,dep+);
    }
}
void Dfs(int x){
    f[x][]=v[x];
    for(int i=;i<=son[x][];i++){
        int v=son[x][i];Dfs(v);
        for(int j=m-c[x];j>=;j--)
            for(int k=;k<j;k++)
                    f[x][j]=max(f[x][j],f[x][j-k]+f[v][k]);
    }
}
int main()
{
    n=init();m=init();m++;
    int fa;
    for(int i=;i<=n;i++){
        fa=init();v[i]=init();
        Add(fa,i);
    }
    Get(,);Dfs();
    printf("%d\n",f[][m]);
    return ;
}