scanf bre discuss 特殊 希望 rip str sort 如果

4027: [HEOI2015]兔子與櫻花

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 1159 Solved: 664
[Submit][Status][Discuss]

Description

很久很久之前，森林裏住著一群兔子。有一天，兔子們突然決定要去看櫻花。兔子們所在森林裏的櫻花樹很特殊。櫻花樹由n個樹枝分叉點組成，編號從0到n-1，這n個分叉點由n-1個樹枝連接，我們可以把它看成一個有根樹結構，其中0號節點是根節點。這個樹的每個節點上都會有一些櫻花，其中第i個節點有c_i朵櫻花。櫻花樹的每一個節點都有最大的載重m，對於每一個節點i，它的兒子節點的個數和i節點上櫻花個數之和不能超過m，即son(i) + c_i <= m，其中son(i)表示i的兒子的個數，如果i為葉子節點，則son(i) = 0

現在兔子們覺得櫻花樹上節點太多，希望去掉一些節點。當一個節點被去掉之後，這個節點上的櫻花和它的兒子節點都被連到刪掉節點的父節點上。如果父節點也被刪除，那麽就會繼續向上連接，直到第一個沒有被刪除的節點為止。 現在兔子們希望計算在不違背最大載重的情況下，最多能刪除多少節點。 註意根節點不能被刪除，被刪除的節點不被計入載重。

Input

第一行輸入兩個正整數，n和m分別表示節點個數和最大載重

第二行n個整數c_i，表示第i個節點上的櫻花個數 接下來n行，每行第一個數k_i表示這個節點的兒子個數，接下來k_i個整數表示這個節點兒子的編號

Output

一行一個整數，表示最多能刪除多少節點。

Sample Input

10 4
0 2 2 2 4 1 0 4 1 1
3 6 2 3
1 9
1 8
1 1
0
0
2 7 4
0
1 5
0

Sample Output

4

HINT

對於100%的數據，1 <= n <= 2000000, 1 <= m <= 100000, 0 <= c_i <= 1000

數據保證初始時，每個節點櫻花數與兒子節點個數之和大於0且不超過m

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5
 
 #include<vector>
 6 using namespace std;
 7 vector<int>edge[2000010];
 8 int n,m,ans;
 9 int w[2000010];
10 bool cmp(int aa,int bb){
11     return w[aa]<w[bb];
12 }
13 void dfs(int u){
14     for(int i=0;i<edge[u].size();i++) dfs(edge[u][i]);
15     sort(edge[u].begin(),edge[u].end(),cmp);
16     w[u]+=edge[u].size(); 
17     for(int i=0;i<edge[u].size();i++)
18         if(w[u]+w[edge[u][i]]-1<=m){
19             w[u]+=w[edge[u][i]]-1;
20             ans++;
21         }
22         else break;
23 }
24 int main(){
25     scanf("%d%d",&n,&m);
26     for(int i=0;i<n;i++) scanf("%d",&w[i]);
27     int k,a;
28     for(int i=0;i<n;i++){
29         scanf("%d",&k);
30         for(int j=1;j<=k;j++){
31             scanf("%d",&a);
32             edge[i].push_back(a);
33         }
34     }
35     dfs(0);
36     printf("%d",ans);
37     return 0;
38 }

dfs+貪心 BZOJ4027 [HEOI2015] 兔子與櫻花