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  = 1000000007;
const int inf = 0x3f3f3f3f;
const int INF = 2000000000;
/**************************************************************************/ 
 
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+1);
    }
}

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