CodeForces 414D （贪心）

problem Mashmokh and Water Tanks

题目大意

　　给你一棵树，k升水，p块钱，进行一次游戏。

　　在游戏进行前，可以在任意个节点上放置1升水（总数不超过k）

　　游戏进行若干轮，每轮游戏开放所有节点，可以选择若干个节点关闭，代价为该节点的水数量。然后所有未关闭的节点的水流向它的父亲(不会连续移动)。最后，根节点中的水会被取走，水的数量为该轮游戏的盈利。

　　求盈利最大的某轮游戏的盈利。

解题分析

　　在放置水的选择上，应该尽量选择深度相邻的节点，即将所有节点按照深度排序后，所选择放水的节点应该是连续的一段。

　　考虑选择某段区间后，所需要花费的钱。

　　假设深度范围为 [l , r] ，某个深度的点为 a[i] ，则花费钱为sigma(a[i]*(r-i))

　　用两个指针进行扫描即可。

参考程序

 #include <map>
 #include <set>
 #include <stack>
 #include <queue>
 #include <cmath>
 #include <ctime>
 #include <string>
 #include <vector>
 #include <cstdio>
 #include <cstdlib>
 #include <cstring>
 #include <cassert>
 #include <iostream>
 #include <algorithm>
 #pragma comment(linker,"/STACK:102400000,102400000")
 using namespace std;

 #define V 100008
 #define E 200008
 #define LL long long
 #define lson l,m,rt<<1
 #define rson m,r+1,rt<<1|1
 #define clr(x,v) memset(x,v,sizeof(x));

 const int mo  = ;
 const int inf = 0x3f3f3f3f;
 const int INF = ;
 /**************************************************************************/ 

 int n,water,coin,d;
 int h[V];

 struct line{
     int u,v,nt;
 }eg[E];
 int lt[V],sum;

 void add(int u,int v){
     eg[++sum]=(line){u,v,lt[u]}; lt[u]=sum;
 }

 void dfs(int u,int fa,int dep){
     h[u]=dep;
     for (int i=lt[u];i;i=eg[i].nt){
         int v=eg[i].v;
         if (v!=fa) dfs(v,u,dep+);
     }
 }

 int main(){
     clr(lt,); sum=;
     scanf("%d %d %d",&n,&water,&coin);
     for (int i=;i<n;i++){
         int u,v;
         scanf("%d %d",&u,&v);
         add(u,v);
         add(v,u);
     }
     dfs(,,);
     sort(h+,h+n+);
     int i=,j=,ans=;
     long long cost=;
     while (i<n && j<n){
         j++;
         if (h[j]!=h[j-]) cost+=j-i;
         while (cost>coin){
             cost-=h[j]-h[i];
             i++;
         }
         ans=max(ans,j-i+);
     }
     printf("%d\n",min(ans,water));
 }