DP——樹形DP——HDOJ1620
阿新 • • 發佈:2018-12-24
題目連結:http://acm.hdu.edu.cn/showproblem.php?pid=1520
大犇解題報告:http://blog.csdn.net/woshi250hua/article/details/7641589#comments
用陣列模擬實現鄰接關係:
//http://acm.hdu.edu.cn/showproblem.php?pid=1520
#include <iostream>
#include <cstdio>
#include <cstring>
//using namespace std;
#define max(a,b) (a)>(b) ? (a) : (b)
int tree[6600][20]; // 這個陣列的第二位應該開到6000,
// 防止極限資料,但執行時間會急劇增加。
//這hdoj的測試平臺上,開到20就可以AC了。
int vis[6699],dp[6600][2],num[6600];
void f(int u)
{
if(vis[u])
return;
else {
vis[u]=1;
// for (int j=tree[u][0];j>=1;j--){
if(!vis[tree[u][j]]){
f(tree[u][j]);
int w=tree[u][j];
dp[u][1]+=dp[w][0];
dp[u][0]+=max(dp[w][0],dp[w][1]);
}
}
}
dp[u][1]+=num[u];
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF){
for (int j=1;j<=n;j++)
scanf("%d", &num[j]);
memset(vis, 0,sizeof(vis));
memset(tree,0,sizeof(tree));
memset(dp,0,sizeof(dp));
int a,b;
while(scanf("%d%d",&a,&b),a+b){
tree[a][++tree[a][0]]=b;
tree[b][++tree[b][0]]=a;
}
f(1);
// std::cout << max(dp[1][1],dp[1][0])<<std::endl;
printf("%d\n", max(dp[1][1],dp[1][0]));
}
}
用二維vector實現鄰接關係:
//http://acm.hdu.edu.cn/showproblem.php?pid=1520
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
#define max(a,b) (a)>(b) ? (a) : (b)
//int tree[6600][20]; // 這個陣列的第二位應該開到6000,
// // 防止極限資料,但執行時間會急劇增加。
// //這hdoj的測試平臺上,開到20就可以AC了。
int vis[6699],dp[6600][2],num[6600];
vector<vector< int> > tree;
void f(int u)
{
if(vis[u])
return;
else {
vis[u]=1;
// for (int j=tree[u][0];j>=1;j--){
for (int j=tree[u].size()-1;j>=0;j--){
if(!vis[tree[u][j]]){
f(tree[u][j]);
int w=tree[u][j];
dp[u][1]+=dp[w][0];
dp[u][0]+=max(dp[w][0],dp[w][1]);
}
}
}
dp[u][1]+=num[u];
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF){
for (int j=1;j<=n;j++)
scanf("%d", &num[j]);
memset(vis, 0,sizeof(vis));
// memset(tree,0,sizeof(tree));
tree.clear();
tree.resize(n+1);
memset(dp,0,sizeof(dp));
int a,b;
while(scanf("%d%d",&a,&b),a+b){
// tree[a][++tree[a][0]]=b;
// tree[b][++tree[b][0]]=a;
tree[a].push_back(b);
tree[b].push_back(a);
}
f(1);
// std::cout << max(dp[1][1],dp[1][0])<<std::endl;
printf("%d\n", max(dp[1][1],dp[1][0]));
}
}