poj 2486 Apple Tree(트리 백팩)
2181 단어 동적 기획동적 계획 - 트리 dp
나무 한 그루를 주면 나무의 각 노드마다 대응하는 수량의 사과가 있다. 한 사람은 어느 지점에서 출발해서든 k보 이내에서 가장 많은 사과를 얻을 수 있는 방법을 물어볼 수 있다.
문제 풀이:
분석에 의하면 이 문제는 나무를 걷는 경로에 대해 왕복과 왕복하지 않는 상황이 있고 한 길을 여러 번 반복할 수 있다는 것을 알아차렸다.
그러면 이러한 상태 dp[root][j][st]뿌리 노드 root의 나무를 정의할 수 있습니다. j보 상태가 st일 때 얻을 수 있는 최대 사과 수입니다.st=0、1. st=0은 루트 노드로 돌아가지 않고 st=1의 반대를 나타낸다.
그러면 상태 방정식이 나온다.
dp[u][j+2][1]=max(dp[u][j+2][1],dp[u][j-k][1]+dp[v][k][1]); dp[u][j+1][0]=max(dp[u][j+1][0],dp[u][j-k][1]+dp[v][k][0]); dp[u][j+2][0]=max(dp[u][j+2][0],dp[u][j-k][0]+dp[v][k][1]);
여기에서 초기화에 주의해야 합니다. 코드를 보십시오.
#include
#include
#include
#include
#include
using namespace std;
typedef __int64 lld;
#define oo 0x3f3f3f3f
#define maxn 105
#define maxm 205
int dp[maxn][maxm][2];
struct EDGE
{
int v,next;
}E[maxn<<1];
int head[maxn],tol;
int val[maxn];
int n,K;
void inst()
{
memset(head,-1,sizeof head);
tol=0;
memset(dp,0,sizeof dp);
}
void add_edge(int u,int v)
{
E[tol].v=v;
E[tol].next=head[u];
head[u]=tol++;
}
void tree_dp(int u,int pre)
{
for(int i=0;i<=K;i++)
dp[u][i][0]=dp[u][i][1]=val[u];//
for(int i=head[u];i!=-1;i=E[i].next)
{
int v=E[i].v;
if(v==pre) continue;
tree_dp(v,u);
for(int j=K;j>=0;j--)
for(int k=0;k<=j;k++)
{
dp[u][j+2][1]=max(dp[u][j+2][1],dp[u][j-k][1]+dp[v][k][1]);
dp[u][j+1][0]=max(dp[u][j+1][0],dp[u][j-k][1]+dp[v][k][0]);//
dp[u][j+2][0]=max(dp[u][j+2][0],dp[u][j-k][0]+dp[v][k][1]);//
}
}
}
int main()
{
int u,v;
while(scanf("%d%d",&n,&K)!=EOF)
{
inst();
for(int i=1;i<=n;i++)
scanf("%d",&val[i]);
for(int i=1;i<=n-1;i++)
{
scanf("%d%d",&u,&v);
add_edge(u,v);
add_edge(v,u);
}
tree_dp(1,-1);
printf("%d
",max(dp[1][K][0],dp[1][K][1]));
}
return 0;
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
01 가방, 완전 가방, 다중 가방 dp(동적 기획 입문 dp)01 가방은 2진법으로 직접 표시할 수 있지만 데이터 양이 너무 많으면 시간을 초과하는 것이 폭력이다.01 가방의 사상은 바로 이 물품에 대해 내가 넣은 가치가 큰지 안 넣은 가치가 큰지 비교하여 방정식 f[i][v...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.