1. 程式人生 > >重建道路

重建道路

getchar code rep || print i++ str i節點 計算

傳送門

我們來看一道比較可做的樹形DP。

現在這個數據規模的樹形DP都是可以直接n3暴力轉移過掉的呀……

不過這個狀態比較特殊,我們用dp[i][j]表示i節點在子樹中保留j個節點所需要刪去的最小邊數。

那麽轉移方程就是,dp[i][j] = min(dp[i][j],dp[i][j-k] + dp[v][k] - 1);其中那個-1是節點i和節點v之間的連邊,那個是要保留的orz,他之前會被重復計算。

所以我們直接這樣暴力dp即可,一開始的初始化是dp[i][1] = son[i],son表示子節點個數。

看一下代碼。

#include<cstdio>
#include<algorithm>
#include
<cstring> #include<cmath> #include<queue> #include<set> #define rep(i,a,n) for(int i = a;i <= n;i++) #define per(i,n,a) for(int i = n;i >= a;i--) #define enter putchar(‘\n‘) using namespace std; typedef long long ll; const int M = 10005; const int INF = 1000000009; int read() {
int ans = 0,op = 1; char ch = getchar(); while(ch < 0 || ch > 9) { if(ch == -) op = -1; ch = getchar(); } while(ch >= 0 && ch <= 9) { ans *= 10; ans += ch - 0; ch = getchar(); } return ans * op; } struct
edge { int next,to,v; }e[10005]; int son[10005],n,p,dp[3005][3005],x,y,ecnt,head[10005],sum[10005]; void add(int x,int y) { e[++ecnt].to = y; e[ecnt].next = head[x]; head[x] = ecnt; } void dfs(int x,int fa) { sum[x] = 1; bool flag = 1; for(int i = head[x];i;i = e[i].next) { int k = e[i].to; if(k == fa) continue; flag = 0; dfs(k,x); sum[x] += sum[k]; per(j,sum[x],0) rep(p,1,j-1) dp[x][j] = min(dp[x][j],dp[x][j-p] + dp[k][p] - 1); } if(flag) { dp[x][1] = 0; sum[x] = 1; return; } } int main() { n = read(),p = read(); rep(i,1,n-1) x = read(),y = read(),add(x,y),add(y,x),son[x]++; memset(dp,127/3,sizeof(dp)); rep(i,1,n) dp[i][1] = son[i]; dfs(1,0); int ans = dp[1][p]; rep(i,2,n) ans = min(ans,dp[i][p] + 1); printf("%d\n",ans); return 0; }

重建道路