loj10153 二叉苹果树

Posted 2020-12-31 水题收割者

传送门

分析

一道简单的树型dp，我们用dp[i][j]记录考虑到第i个点，保留了j个树枝的最多苹果数，然后dfs求解即可。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define pb push_back
#define mp make_pair
vector<pair<int,int> >v[110];
pair<int,int>son[2][110];
int dp[110][110],n,m;
inline void dfs(int x,int fa){
      int cnt=-1;
      for(int i=0;i<v[x].size();i++)
        if(v[x][i].first!=fa){
          son[++cnt][x]=v[x][i];
          dfs(v[x][i].first,x);
        }
      return;
}
inline void getdp(int x){
      if(!son[0][x].first)return;
      getdp(son[0][x].first);
      getdp(son[1][x].first);
      for(int i=0;i<n-2;i++)
        for(int j=0;j<=i;j++)
          dp[x][i+2]=max(dp[x][i+2],dp[son[0][x].first][j]+
          dp[son[1][x].first][i-j]+son[0][x].second+son[1][x].second);
      for(int i=0;i<n-1;i++)
        for(int j=0;j<=i;j++)
          dp[x][i+1]=max(dp[x][i+1],max(dp[son[0][x].first][j]+
          son[0][x].second,dp[son[1][x].first][i-j]+son[1][x].second));
      return;
} 
int main(){
      int i,j,k,x,y;
      scanf("%d%d",&n,&m);
      for(i=1;i<n;i++){
          scanf("%d%d%d",&x,&y,&k);
          v[x].pb(mp(y,k));
          v[y].pb(mp(x,k));
      }
      memset(dp,0,sizeof(dp));
      dfs(1,0);
      getdp(1);
      printf("%d
",dp[1][m]);
      return 0;
}