二叉苹果树(树型DP+背包)
生活随笔
收集整理的這篇文章主要介紹了
二叉苹果树(树型DP+背包)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
?
二叉蘋果樹
有一棵蘋果樹,如果樹枝有分叉,一定是分2叉(就是說沒有只有1個(gè)兒子的結(jié)點(diǎn))。這棵樹共有N個(gè)結(jié)點(diǎn)(葉子點(diǎn)或者樹枝分叉點(diǎn)),編號(hào)為1-N,樹根編號(hào)一定是1。
我們用一根樹枝兩端連接的結(jié)點(diǎn)的編號(hào)來描述一根樹枝的位置。下面是一顆有4個(gè)樹枝的樹:
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 2 ? 5
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?\ ?/
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?3 ?4
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? \ ?/
程序名:apple
輸入格式:
? ? ? ? ?第1行2個(gè)數(shù),N和Q(1<=Q<= N,1<N<=100)。
? ? ? ? ?N表示樹的結(jié)點(diǎn)數(shù),Q表示要保留的樹枝數(shù)量。接下來N-1行描述樹枝的信息。
? ? ? ? ?每行3個(gè)整數(shù),前兩個(gè)是它連接的結(jié)點(diǎn)的編號(hào)。第3個(gè)數(shù)是這根樹枝上蘋果的數(shù)量。
? ? ? ? ?1 3 1
? ? ? ? ?1 4 10
? ? ? ? ?2 3 20
? ? ? ? ?3 5 20
? 輸入樣例:
有一棵蘋果樹,如果樹枝有分叉,一定是分2叉(就是說沒有只有1個(gè)兒子的結(jié)點(diǎn))。這棵樹共有N個(gè)結(jié)點(diǎn)(葉子點(diǎn)或者樹枝分叉點(diǎn)),編號(hào)為1-N,樹根編號(hào)一定是1。
我們用一根樹枝兩端連接的結(jié)點(diǎn)的編號(hào)來描述一根樹枝的位置。下面是一顆有4個(gè)樹枝的樹:
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? 2 ? 5
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?\ ?/
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?3 ?4
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? \ ?/
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?1
現(xiàn)在這顆樹枝條太多了,需要剪枝。但是一些樹枝上長有蘋果。
給定需要保留的樹枝數(shù)量,求出最多能留住多少蘋果。程序名:apple
輸入格式:
? ? ? ? ?第1行2個(gè)數(shù),N和Q(1<=Q<= N,1<N<=100)。
? ? ? ? ?N表示樹的結(jié)點(diǎn)數(shù),Q表示要保留的樹枝數(shù)量。接下來N-1行描述樹枝的信息。
? ? ? ? ?每行3個(gè)整數(shù),前兩個(gè)是它連接的結(jié)點(diǎn)的編號(hào)。第3個(gè)數(shù)是這根樹枝上蘋果的數(shù)量。
? ? ? ? ?每根樹枝上的蘋果不超過30000個(gè)。
?輸出格式:
? ? ? ? ?一個(gè)數(shù),最多能留住的蘋果的數(shù)量。?輸入樣例:
? ? ? ? ?5 2? ? ? ? ?1 3 1
? ? ? ? ?1 4 10
? ? ? ? ?2 3 20
? ? ? ? ?3 5 20
? 輸入樣例:
? ? ? ? ? 21
? 解題思路:樹型DP+背包求解。
? ? ? ? ? ? ? ? ? ? ??f(i, j) 表示子樹i,保留j個(gè)節(jié)點(diǎn)(注意是節(jié)點(diǎn))的最大權(quán)值。每條邊的權(quán)值,把它看作是連接的兩個(gè)節(jié)點(diǎn)中的兒子節(jié)點(diǎn)的權(quán)值。
? ? ? ? ? ? ? ? ? ? ? 那么,就可以對(duì)所有i的子樹做分組背包,即每個(gè)子樹可以選擇1,2,...j-1條邊分配給它。
? ? ? ? ? ? ? ? ? ? ? 狀態(tài)轉(zhuǎn)移為:
? ? ? ? ? ? ? ? ? ? ? ?f(i, j) = max{ max{f(i, j-k) + f(v, k) | 1<=k<j} | v是i的兒子}?
? ? ? ? ? ? ? ? ? ? ? ?ans = f(1, q+1)
? 代碼如下:? ? ? ? ? ?
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #include<vector> using namespace std; #define MAX 105 #define INF 999999999 #define MP make_pair typedef pair<int,int> PII; vector<PII> adj[MAX]; int tot[MAX],f[MAX][MAX]; int max(int a,int b) {return a>b?a:b; } int DFS(int u,int ff) {tot[u]=1;int i,j,k;for(i=0;i<adj[u].size();i++){int v=adj[u][i].first;if(v==ff)continue;tot[u]+=DFS(v,u);}for(i=0;i<adj[u].size();i++){int v=adj[u][i].first;int w=adj[u][i].second;if(v==ff)continue;for(j=tot[u];j>1;j--){for(k=1;(k<j)&&(k<=tot[v]);k++)f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]+w);}}return tot[u]; } int main() {int i;int n,p,u,v,w;while(~scanf("%d%d",&n,&p)){for(i=0;i<MAX;i++)adj[i].clear();for(i=1;i<n;i++){scanf("%d%d%d",&u,&v,&w);adj[u].push_back(MP(v,w));adj[v].push_back(MP(u,w));}memset(f,0,sizeof(f));DFS(1,-1);printf("%d\n",f[1][p+1]);}return 0; }總結(jié)
以上是生活随笔為你收集整理的二叉苹果树(树型DP+背包)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: NYOJ 420 P次方求和
- 下一篇: 阿里晓斌:如何做好技术 Team Lea