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

luogu P1272 重建道路

mes memset tchar getchar() splay queue gif tac pre

嘟嘟嘟

這好像是一種樹上背包。

我們令dp[i][j] 表示在 i 所在的子樹中(包括節點 i)分離出一個大小為 j 的子樹最少需割多少條邊。

那麽轉移方程就是

  dp[u][j] = min(dp[u][j], dp[u][j - k] + dp[v][k] - 1) (v是u的一個兒子)

理解起來就是在u所在子樹中切 j 個節點,等於在u中切 j - k 個節點加上在v所在子樹中切 k 個節點所需要切的邊數之和。又因為切出來的這兩部分要合並得到一個節點數為 j 的,所以要減1.

技術分享圖片
 1 #include<cstdio>
 2 #include<iostream>
 3
#include<cmath> 4 #include<algorithm> 5 #include<cstring> 6 #include<cstdlib> 7 #include<cctype> 8 #include<vector> 9 #include<stack> 10 #include<queue> 11 using namespace std; 12 #define enter puts("") 13 #define space putchar(‘ ‘) 14 #define Mem(a, x) memset(a, x, sizeof(a)) 15
#define rg register 16 typedef long long ll; 17 typedef double db; 18 const int INF = 0x3f3f3f3f; 19 const db eps = 1e-8; 20 const int maxn = 155; 21 inline ll read() 22 { 23 ll ans = 0; 24 char ch = getchar(), last = ; 25 while(!isdigit(ch)) {last = ch; ch = getchar();} 26 while(isdigit(ch)) {ans = ans * 10
+ ch - 0; ch = getchar();} 27 if(last == -) ans = -ans; 28 return ans; 29 } 30 inline void write(ll x) 31 { 32 if(x < 0) x = -x, putchar(-); 33 if(x >= 10) write(x / 10); 34 putchar(x % 10 + 0); 35 } 36 37 int n, p; 38 vector<int> v[maxn]; 39 int du[maxn], siz[maxn]; 40 int dp[maxn][maxn]; 41 42 void dfs(int now) 43 { 44 siz[now] = 1; 45 for(int i = 0; i < (int)v[now].size(); ++i) 46 { 47 dfs(v[now][i]); 48 siz[now] += siz[v[now][i]]; 49 for(int j = siz[now]; j >= 0; --j) 50 for(int k = 1; k < j; ++k) 51 dp[now][j] = min(dp[now][j], dp[now][j - k] + dp[v[now][i]][k] - 1); 52 } 53 } 54 55 int main() 56 { 57 n = read(); p = read(); 58 for(int i = 1; i < n; ++i) 59 { 60 int x = read(), y = read(); 61 v[x].push_back(y); 62 du[x]++; 63 } 64 for(int i = 0; i <= n; ++i) 65 for(int j = 0; j <= n; ++j) dp[i][j] = INF; 66 for(int i = 1; i <= n; ++i) dp[i][1] = du[i]; 67 dfs(1); 68 int ans = dp[1][p]; //本來就是根節點 69 for(int i = 2; i <= n; ++i) ans = min(ans, dp[i][p] + 1); //還要切除和父親連的一條邊 70 write(ans); enter; 71 return 0; 72 }
View Code

luogu P1272 重建道路