2019牛客暑假多校训练赛第七场 C题 Governing sand

Governing sand

题目链接：https://ac.nowcoder.com/acm/contest/887/C

题意：给你一片森林，让你砍树，保证最高树的数量大于总数量的一半，给你树的种类n，hi,ci,pi；求最小花费。

数据范围：(1≤n≤1e5),(0<=Ci<=200),(0<=Hi<=1e9)

思路：肯定是O(n*c)的复杂度，如果贪心确定不了树的高度，dp也不行，只能枚举每一个树的高度当做最大高度，计算花费的时候要拿上之前砍掉比它高的树的花费+剩下的最小花费，暴力200就能取剩下的数量了。

代码实现：肯定要预处理，处理按高度从小到大（或者从大到小）的前缀和，后缀和（加的时候方便），然后O(n*c)暴力就行了。

 #include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll pre[200005],fix[200005];
ll num[305];
struct node
{
    ll h,c,p;
} a[200020];
bool cmp(node a,node b)
{
    return a.h<=b.h;
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        memset(num,0,sizeof num);
        map<ll,ll>mp;
        for(int i=1; i<=n; i++)
        {
            scanf("%lld %lld %lld",&a[i].h,&a[i].c,&a[i].p);
            mp[a[i].h]+=a[i].p;
        }
        sort(a+1,a+n+1,cmp);
        // pre[0]=0,fix[n+1]=0;
        for(int i=1; i<=n; i++)
            pre[i]=pre[i-1]+a[i].p*a[i].c;
        for(int j=n; j>=1; j--)
            fix[j]=fix[j+1]+a[j].p*a[j].c;
        ll ans=1e18;
        for(int i=1; i<=n; i++)
        {
            ll sum=0;
            ll all=mp[a[i].h]-1;
            for(ll j=200; j>=1; j--)
            {
                if(num[j]<all)
                {
                    sum+=num[j]*j;
                    all-=num[j];
                }
                else
                {
                    sum+=all*j;
                    all=0;
                    break;
                }
            }
            num[a[i].c]+=a[i].p;
            int pos=i+1;
            while(a[pos].h==a[i].h)
                pos++;
            ans=min(ans,pre[i-1]-sum+fix[pos]);

        } cout<<ans<<endl;
    }
}