【贪心题解】 CF799C Fountains

题意

有 n 座喷泉，每座喷泉有一魅力值，可用金币或钻石一种货币来买，从喷泉中选 2 座喷泉，使两座喷泉美丽值总和最大且买得起。

思路

看完题，我们可以轻松想到暴力做法，也就是两重循环枚举每一点并更新答案，但是很明显时间复杂度为  ，是要超时的。

考虑优化一重循环，也就是  或者  算法，这时就可以使用树状数组，以喷泉的金币或钻石数为下标，美丽度为权值建树状数组，然后就相当于每次剩余的钱中间，找到一个最大值，更新答案。

 

 代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=100005;
int n,c,d,cnt1,cnt2,k[maxn];
int ans1=-1,ans2=-1;

struct node{
    int cnt,p;
}a1[maxn],a2[maxn];

int LowBit(int x){
    return x&-x;
}

void UpDate(int x,int y,int val){
    while(x<y){
        k[x]=max(k[x],val);
        x+=LowBit(x);
    }
}

int Sum(int x){
    int ans=0;
    while(x>0){
        ans=max(ans,k[x]);
        x-=LowBit(x);
    }
    return ans;
}

int main(){
    scanf("%d%d%d",&n,&c,&d);
    for(int i=0;i<n;i++){
        int b,p;
        char s;
        scanf("%d%d %c",&b,&p,&s);
        if(s=='C'){
            if(p<=c){
                a1[cnt1].cnt=b;
                a1[cnt1++].p=p;
                ans1=max(ans1,b);
            }
        }
        else{
            if(p<=d){
                a2[cnt2].cnt=b;
                a2[cnt2++].p=p;
                ans2=max(ans2,b);
            }
        }
    }
    int ans=0;
    if(ans1!=-1&&ans2!=-1){
        ans=ans1+ans2;
    }
    memset(k,0,sizeof(k));
    for(int i=0;i<cnt1;i++){
        int tot=Sum(c-a1[i].p);
        if(tot>0){
            ans=max(ans,tot+a1[i].cnt);
        }
        UpDate(a1[i].p,maxn,a1[i].cnt);
    }
    memset(k,0,sizeof(k)); 
    for(int i=0;i<cnt2;i++){
        int tot=Sum(d-a2[i].p);
        if(tot>0){
            ans=max(ans,tot+a2[i].cnt);
        }
        UpDate(a2[i].p,maxn,a2[i].cnt);
    }
    printf("%d",ans);
    return 0;
}