【poj 1770】Special Experiment 树形dp

来源:互联网 发布:校园小说改编的网络剧 编辑:程序博客网 时间:2024/05/17 06:35

题意:输入n,m。表示有n个原子和m个光子,接下来有是n个原子的能量和m个光子的能量,如果有任意两个原子之差等于一个光子就会爆炸啦啦啦,叫你选出不会爆炸的最大的原子能量


这道题感觉有点马后炮的感觉,discuss里面说不会出现环,所以就放心的将两个会爆炸的点连一条边然后就变成了一个经典问题:选父亲就不要儿子,求最大值,好像叫最大独立子集神马的对吧,不记得了,反正就那么一回事

#include<cstdio>#include<cstring>#include<iostream>#include<vector>#include<cstdlib>#include<cmath>#define maxn 210#define PB push_back#define Clear(x) memset(x,0,sizeof(x))using namespace std;int n,m,a[maxn],vis[maxn],f[maxn][2];bool b[2000020];vector<int>g[maxn];void dfs(int u){vis[u]=1;for(int i=0;i<g[u].size();i++){int v=g[u][i];if(vis[v])continue;dfs(v);f[u][0]+=max(f[v][1],f[v][0]);f[u][1]+=f[v][0];}f[u][1]+=a[u];}int main(){while(scanf("%d%d",&n,&m)!=EOF&&m+n){Clear(a);Clear(f);Clear(b);Clear(vis);for(int i=1;i<=n;i++)g[i].clear();for(int i=1;i<=n;i++)scanf("%d",a+i);for(int x,i=1;i<=m;i++){scanf("%d",&x);b[x]=true;}for(int i=1;i<=n;i++){for(int j=1;j<=n;j++){if(i==j)continue;if(b[abs(a[i]-a[j])])g[i].PB(j);}}int ans=0;for(int i=1;i<=n;i++){if(!vis[i]){dfs(i);ans+=max(f[i][0],f[i][1]);}}printf("%d\n",ans);}return 0;}

0 0
原创粉丝点击