BZOJ1063 NOI2008 道路设计 树形DP

题目传送门：

BZOJ

题意精简版：给出一棵树，在一种方案中可以将树的若干链上的所有边的边权改为$0$，但需要保证任意两条链之间没有交点。问最少的一种方案，使得从根节点到其他节点经过的边的边权和的最大值最小，并求出方案数。$N \leq 10^5$

---

 真心火题，现在我还在想是哪个神人推出的这个DP方程

关于原题中的无解情况直接判断$M==N-1$是否成立即可，因为根据题目要求，最后生成出来的一定会是一棵树。

先考虑第二问，设$f_{i,j,k}$表示在子树$i$内，最大的边权和不超过$j$，点$i$与其$k$个儿子在一条链上的方案数。首先可以知道$k=0,1,2$，因为任意两条链之间没有交点。

接下来我们考虑$j$的取值。手玩可以发现，当一棵树是三叉树的时候，才会有一个边权的贡献，而当整棵树是深度为$h$的满二叉树的时候，才会有$h$的最大边权，而$\sum\limits_{i=0}^{10} 3^i = 8.6 \times 10^4$，所以可以知道答案会小于等于10，所以$j$的有效取值范围就在$[0,10]$中间，第一问的答案自然也就很小。

接下来考虑转移。根据乘法原理，我们可以得到这样子的式子：

$$f_{i,j,2}=f_{i,j,1} \times (f_{soni,j,0} + f_{soni,j,1}) + f_{i,j,2} \times (f_{soni,j-1,0} + f_{soni,j-1,1} + f_{soni,j-1,2})$$

$$f_{i,j,1}=f_{i,j,0} \times (f_{soni,j,0} + f_{soni,j,1}) + f_{i,j,1} \times (f_{soni,j-1,0} + f_{soni,j-1,1} + f_{soni,j-1,2})$$

$$f_{i,j,0}=f_{i,j,0} \times (f_{soni,j-1,0} + f_{soni,j-1,1} + f_{soni,j-1,2})$$

其中$soni$表示$i$的某个儿子。边界是$f_{i,j,0}=1$，其余为$0$

每一个点统计儿子的所有转移即可，复杂度为$O(33N)$。

至于第一问，可以从小到大枚举dp的第$2$维，最小的在$1$号点上方案数大于$0$的数就是答案。

注意：因为原题需要$mod\,Q$，所以如果答案$mod\,Q==0$会出现错误，推荐在某个数与$mod \, Q = 0$时将其修改为$Q$而不是$0$。

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 
 4 inline int read(){
 5     int a = 0;
 6     char c = getchar();
 7     while(!isdigit(c))
 8         c = getchar();
 9     while(isdigit(c)){
10         a = (a << 3) + (a << 1) + (c ^ '0');
11         c = getchar();
12     }
13     return a;
14 }
15 
16 const int MAXN = 100010;
17 struct Edge{
18     int end , upEd;
19 }Ed[MAXN << 1];
20 int head[MAXN] , fa[MAXN] , cntEd , N , M , Q;
21 long long dp[MAXN][11][3];
22 bool vis[MAXN];
23 
24 inline void addEd(int a , int b){
25     Ed[++cntEd].end = b;
26     Ed[cntEd].upEd = head[a];
27     head[a] = cntEd;
28 }
29 
30 inline int get(long long x){
31     return x != 0 && x % Q == 0 ? Q : x % Q;
32 }
33 
34 void dfs(int now){
35     vis[now] = 1;
36     for(int i = 0 ; i <= 10 ; i++)
37         dp[now][i][0] = 1;
38     for(int i = head[now] ; i ; i = Ed[i].upEd)
39         if(!vis[Ed[i].end]){
40             fa[Ed[i].end] = now;
41             dfs(Ed[i].end);
42         }
43     for(int j = 1 ; j <= 10 ; j++){
44         long long k = get(dp[now][j - 1][0] + dp[now][j - 1][1] + dp[now][j - 1][2]) , l = get(dp[now][j][0] + dp[now][j][1]);
45         dp[fa[now]][j][2] = get(dp[fa[now]][j][1] * l + dp[fa[now]][j][2] * k);
46         dp[fa[now]][j][1] = get(dp[fa[now]][j][1] * k + dp[fa[now]][j][0] * l);
47         dp[fa[now]][j][0] = get(dp[fa[now]][j][0] * k);
48     }
49     dp[fa[now]][0][2] = get(dp[fa[now]][0][1] * (dp[now][0][0] + dp[now][0][1]));
50     dp[fa[now]][0][1] = get(dp[fa[now]][0][0] * (dp[now][0][0] + dp[now][0][1]));
51     dp[fa[now]][0][0] = 0;
52 }
53 
54 int main(){
55     N = read();
56     M = read();
57     if(M != N - 1){
58         cout << "-1\n-1";
59         return 0;
60     }
61     Q = read();
62     for(int i = 1 ; i <= M ; i++){
63         int a = read() , b = read();
64         addEd(a , b);
65         addEd(b , a);
66     }
67     dfs(1);
68     for(int i = 0 ; i <= N ; i++){
69         int sum = dp[1][i][0] + dp[1][i][1] + dp[1][i][2];
70         if(sum){
71             cout << i << endl << sum % Q;
72             return 0;
73         }
74     }
75 }

 

转载于:https://www.cnblogs.com/Itst/p/9839787.html