poj3342 poj2342 树形dp入门

poj3342

最近做树形dp，渣渣被虐成喳喳喳辣，真是不开森。。。。

还是适合做基础入门题。。。╮(╯▽╰)╭

题意：比较好理解。就是一棵树，选m个点，儿子和父亲不能同时选。问最多能选多少个（题目中说的是上级，当然我这里省略就是用父亲节点和孩子结点表示啦~~）

其实dp的内容很好想，d[i][0]表示第i个点不选的结果，d[i][1]表示第i个点选的结果，转移方程d[i][0]=sigma(max(d[j][0],d[j][1])) ,就是儿子可选可不选，选出最大的就ok

                                                                                                                                                                   d[i][1]=sigma(d[j][0]) 父亲选了，当然儿子就不能选。

主要问题是建图，我可能还是不太会，因为这里用的map和vector 。

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<vector>#include<map>using namespace std;#define maxn 220vector<int>edge[maxn]; //记录边的信息 edge[i][j]表示以i为起点的第j条边的终点的序列号int d[maxn][2];void dfs(int u){d[u][0]=0;d[u][1]=1;for(int i=0;i<edge[u].size();i++){int v=edge[u][i];dfs(v);d[u][0]+=max(d[v][0],d[v][1]);d[u][1]+=d[v][0];}}int main(){//freopen("q.in","r",stdin);int i,j,n,tot;//tot记录节点数map<string,int> mp;char bos[110],a[110],b[110];while(~scanf("%d",&n) && n){tot=0;for(i=0;i<maxn;i++)edge[i].clear();mp.clear();memset(d,0,sizeof(d));scanf("%s",bos);mp[bos]=tot++;for(i=1;i<n;i++){scanf("%s%s",a,b);if(mp.find(a)==mp.end())mp[a]=tot++;if(mp.find(b)==mp.end())mp[b]=tot++;edge[mp[b]].push_back(mp[a]);  //形成一条边 }dfs(0);bool flag;for(i=0;i<n;i++){flag=true;if(d[i][0]>d[i][1]){for(j=0;j<edge[i].size();j++){if(d[edge[i][j]][0]==d[edge[i][j]][1]){flag=false;break;}}}if(!flag)break;}cout<<max(d[0][0],d[0][1])<<" ";if(d[0][0]==d[0][1] || !flag)cout<<"No"<<endl;else cout<<"Yes"<<endl;}} 

poj2342 

做完上一题，做这个他觉得思路更清晰了，不过还是错了一点，就是这个题的根节点没有直接给出，还要自己找。我刚开始是觉得根节点不会出现在孩子学列的那一排，也就是程序中的所有的a,然后把所有的a保存起来，拍一下序，看缺谁，谁就是根节点。但是WA。。。

后来看了别人的方法，是孩子父亲保存在一起，一起找，另开了个数组，然后才A的。。。。。。。

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<vector>#include<algorithm> #include<map>using namespace std;#define maxn 6002vector<int> edge[maxn];int d[maxn][2];int w[maxn];int fa[maxn];void dfs(int u){int i,j;d[u][0]=0;d[u][1]=w[u];for(i=0;i<edge[u].size();i++){int v=edge[u][i];dfs(v);d[u][0]+=max(d[v][0],d[v][1]);d[u][1]+=d[v][0];}}int main(){//freopen("q.in","r",stdin);int n,i,j;int sta=0;int a,b;memset(d,0,sizeof(d));memset(w,0,sizeof(w));memset(fa,0,sizeof(fa)); scanf("%d",&n);for(i=1;i<=n;i++)scanf("%d",&w[i]);while(scanf("%d%d",&a,&b)==2 && a+b){fa[a]=b;edge[b].push_back(a); //还需要找到根节点  根节点肯定不会出现在孩子的序列当中   }for(i=1;i<=n;i++)if(!fa[i]){sta=i;break;} //根节点刚开始出了点问题。。。。贡献了WA.... //cout<<sta<<endl;dfs(sta);int ans=max(d[sta][0],d[sta][1]);if(ans<0)cout<<ans<<endl;else cout<<ans<<endl;}