当前位置 : 主页 > 编程语言 > java >

HDOJ 1770 - 阅读理解...树形DP

来源:互联网 收集:自由互联 发布时间:2022-08-15
题意: 一个能量E可以通过吸收某个光子的能量变成E1或者释放某个光子的能量变成E2...并且任意两个能量的转化路径至多一条...现在有一堆能量,有一堆光子...如果某个能量与某个光子做


  题意: 一个能量E可以通过吸收某个光子的能量变成E1或者释放某个光子的能量变成E2...并且任意两个能量的转化路径至多一条...现在有一堆能量,有一堆光子...如果某个能量与某个光子做直接运算(加上其能量或者减去)会等于一个已经存在的能量...那么就会发生危险...问在这堆能量中..不发生危险并且能量和最大为多少...

   由于两个能量的转化路径至多一条..那么可以用树(森林)来表示所有的关系...有冲突的两点做无向边....我之前一直WA就是做成有向边了...

   dp[k][0]代表以k为根的子树,不取k这个点..最多能获得的能量...

   dp[k][1]代表以k为根的子树,取k这个点..最多能获得的能量...

   转化成一个很经典的问题了..


Program:

#include<iostream>
#include<stack>
#include<queue>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<cmath>
#define ll long long
#define oo 1000000007
#define MAXN 205
using namespace std;
vector<int> Tree[MAXN];
int n,m,a[MAXN],dp[MAXN][2];
bool P[1000005],used[MAXN],root[MAXN];
void dfs(int x)
{
int i,num=Tree[x].size();
used[x]=true;
dp[x][0]=0,dp[x][1]=a[x];
for (i=0;i<num;i++)
if (!used[Tree[x][i]])
{
dfs(Tree[x][i]);
dp[x][0]+=max(dp[Tree[x][i]][0],dp[Tree[x][i]][1]);
dp[x][1]+=dp[Tree[x][i]][0];
}
return;
}
int main()
{
int i,x;
while (~scanf("%d%d",&n,&m) && (n || m))
{
for (i=1;i<=n;i++) scanf("%d",&a[i]),Tree[i].clear();
sort(a+1,a+1+n);
memset(P,false,sizeof(P));
P[0]=true;
for (i=1;i<=m;i++) scanf("%d",&x),P[x]=true;
memset(root,true,sizeof(root));
for (i=1;i<=n;i++)
for (x=i+1;x<=n;x++)
if (P[a[x]-a[i]])
Tree[x].push_back(i),Tree[i].push_back(x);
memset(used,false,sizeof(used));
x=0;
for (i=1;i<=n;i++)
if (!used[i])
{
dfs(i);
x+=max(dp[i][0],dp[i][1]);
}
printf("%d\n",x);
}
return 0;
}



网友评论