树形dp系列

1.火车站开饭店

最大独立集裸题

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<algorithm>
 #include<string>
 #define N 200000+20
 using namespace std;
 ];
 ;
 void link(int x,int y){
     nxt[++tot]=head[x];
     to[tot]=y;
     head[x]=tot;
 }
 void dfs(int x,int fa){
     dp[x][]=a[x];//0为在独立集里
     dp[x][]=;//1为不在
     for (int i=head[x];i;i=nxt[i]){
         if (to[i]==fa) continue;
         dfs(to[i],x);
         dp[x][]+=dp[to[i]][];
         dp[x][]+=max(dp[to[i]][],dp[to[i]][]);
     }
 }
 int main(){
     scanf ("%d",&n);
     ;i<=n;++i) scanf ("%d",&a[i]);
     int x,y;
     ;i<n;++i){
         scanf ("%d%d",&x,&y);
         link(x,y),link(y,x);
     }
     dfs(,);
     printf (][],dp[][]));
     ;
   }

2.树的最小染色

考题 当时只打了两个状态 居然用我神奇的调试功力搞了10分?

考完之后愣是想了好久才想通 好像是最小支配集

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#define maxn 300000
#define INF 10000100
using namespace std;
,root=,sum,f[maxn][],ans=,g[maxn],sz[maxn];
void link(int x,int y)
{
    nxt[++tot]=head[x];
    to[tot]=y;
    head[x]=tot;
}
void dfs(int x)
{
    if (!sz[x])//0 靠爸爸 1 靠儿子 2 靠自己
    {
        f[x][]=,f[x][]=INF>>,f[x][]=v[x];
        return ;
    }
        ,fl=;
    for (int i=head[x];i;i=nxt[i])
    {
        int t=to[i];
        dfs(t);
        f[x][]+=min(f[t][],min(f[t][],f[t][]));
        f[x][]+=f[t][];
        ]>=f[t][]) fl=,s2+=f[t][];
        ],s1=min(s1,f[t][]-f[t][]);
    }
    f[x][]+=v[x];
    f[x][]=s2;
    ]+=s1;
}
int main()
{
    int n;
    scanf ("%d",&n);
    ;i<=n;++i)
    {
        int x,y,z,k;
        scanf ("%d%d%d",&x,&y,&z);
        v[x]=y,sz[x]=z;
                ;j<=z;++j)
        {
            scanf ("%d",&k);
            link(x,k);
            g[k]=;
        }
    }
    ;i<=n;++i) if (!g[i]) root=i;
    dfs(root);
    cout<<min(f[root][],f[root][]);
    ;
}    

3.POJ 3659 Cell Phone Network

最小支配集  模板题 另外我发现我真的很喜欢打靠爸爸靠儿子还是靠自己之类的话23333

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<algorithm>
 #include<string>
 #define ll long long
 #define db double
 #define N 20000
 #define inf 1000000000
 using namespace std;
 ];
 ;
 void link(int x,int y){
     nxt[++tot]=head[x];
     to[tot]=y;
     head[x]=tot;
 }
 void dfs(int x,int fa){
     dp[x][]=,dp[x][]=,dp[x][]=;//0属于 1靠儿子 2靠父亲
     ,q=inf;
     for (int i=head[x];i;i=nxt[i]){
         int t=to[i];
         if (t==fa) continue;
         dfs(t,x);
         dp[x][]+=min(dp[t][],min(dp[t][],dp[t][]));
         ]!=inf&&dp[x][]!=inf) dp[x][]+=dp[t][];
         ]=inf;
         ]>=dp[t][]) dp[x][]+=dp[t][],fl=t;
         ]+=dp[t][];
         q=min(q,dp[t][]-dp[t][]);
     }
     ]+=q;
     ])  dp[x][]=inf;
 }
 int main(){
     int n;
     scanf ("%d",&n);
     int x,y;
     ;i<n;++i){
         scanf ("%d%d",&x,&y);
         link(x,y),link(y,x);
     }
     dfs(,);
     printf (][],dp[][]));
     ;
   }

4.POJ 2152 消防站

神题啊qwq 怎么推的出(瘫

围观了一下dalao的论文感觉很有道理又觉得不是很明白

后来翻到:http://blog.csdn.net/loi_dqs/article/details/50878337 在吸取了论文部分内容的基础上很快就明白了

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<algorithm>
 #include<string>
 #include<queue>
 #include<vector>
 #define ll long long
 #define db double
 #define N 5000
 using namespace std;
 ;
 int w[N],d[N],k[N],de[N][N],ans[N],f[N][N];
 struct hh{
     int to,nxt,w;
 }b[N];
 int head[N];
 void link(int x,int y,int w){
     b[++tot].nxt=head[x];
     b[tot].to=y;
     b[tot].w=w;
     head[x]=tot;
 }
 void dis(int x,int y,int fa,int di){
     de[x][y]=min(de[x][y],di);
     for (int i=head[y];i;i=b[i].nxt){
         int v=b[i].to;
         if (v==fa) continue;
         dis(x,v,y,di+b[i].w);
     }
 }
 void dp(int x,int fa){
     for (int i=head[x];i;i=b[i].nxt){
         int t=b[i].to;
         if (t==fa) continue;
         dp(t,x);
     }
     ;i<=n;++i){
         if (de[x][i]>d[x]) continue;
         f[x][i]=w[i];
         for (int j=head[x];j;j=b[j].nxt){
             int k=b[j].to;
             if (k==fa) continue;
             f[x][i]+=min(f[k][i]-w[i],ans[k]);
         }
         ans[x]=min(ans[x],f[x][i]);
     }
 }
 int main(){
     scanf ("%d",&n);
     ;i<=n;++i) scanf ("%d",&w[i]);
     ;i<=n;++i) scanf ("%d",&d[i]);
     int x,y,z;
     ;i<n;++i){
         scanf ("%d%d%d",&x,&y,&z);
         link(x,y,z);link(y,x,z);
     }
     memset(f,0x3f,sizeof(f));
     memset(de,0x3f,sizeof(de));
     memset(ans,0x3f,sizeof(ans));
     ;i<=n;++i) dis(i,i,,);
     dp(,);
     printf (]);
     ;
   }

UPD        2017_07_12

连考两天树形dp我都在鬼混 感觉很不好意思TAT(之前可能是白学了)(那个白学 不是那个白学)

发现了一点小套路?

感觉这一类题经常可以用两个dp的数组分别搞然后分情况转移合并 这样思路比较简单

也可以设三个左右状态互相转移 实现起来会更清爽qwq

然后好像经常设的都是N×M之类的 一般要数据范围别太大才能玩吧

状态大多跟父节点 子节点 子树的个数 在不在某状态中 一类有关qwq

还需多多领悟啊= =

上一篇:Python入门(八)分支与循环


下一篇:Pyhton开源框架(加强版)