51nod - 1163 巧妙的并查集 O(1)维护区间

有N个任务,每个任务有一个最晚结束时间以及一个对应的奖励。在结束时间之前完成该任务,就可以获得对应的奖励。完成每一个任务所需的时间都是1个单位时间。有时候完成所有任务是不可能的,因为时间上可能会有冲突,这需要你来取舍。求能够获得的最高奖励。

从没想过并查集可以这么用
这道题很显然需要做的是贪心把大的先占用掉,然后往小的尽可能塞
那么该怎么塞也应该是贪心的从结束的点开始塞,不行就往前推,这个链式反应的过程如果用并查集来实现,就可以达到接近O(1)的速度
诀窍就是x和p[x]相同时,返回x-1,表示x可以占用,并维护关系

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<string>
#include<vector>
#include<stack>
#include<queue>
#include<set>
#include<map>
#define rep(i,j,k) for(register int i=j;i<=k;i++)
#define rrep(i,j,k) for(register int i=j;i>=k;i--)
#define erep(i,u) for(register int i=head[u];~i;i=nxt[i])
#define iin(a) scanf("%d",&a)
#define lin(a) scanf("%lld",&a)
#define din(a) scanf("%lf",&a)
#define s0(a) scanf("%s",a)
#define s1(a) scanf("%s",a+1)
#define print(a) printf("%lld",(ll)a)
#define enter putchar('\n')
#define blank putchar(' ')
#define println(a) printf("%lld\n",(ll)a)
#define IOS ios::sync_with_stdio(0)
using namespace std;
const int MAXN = 1e5+11;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-7;
typedef long long ll;
const ll MOD = 1e9+7; 
ll read(){
    ll x=0,f=1;register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,p[MAXN];
int find(int x){
    if(x==-1) return x;
    if(p[x]==x) return p[x]=x-1;
    return p[x]=find(p[x]);
}
struct P{
    ll ed,cost;
}a[MAXN];
bool cmp(P a,P b){
    return a.cost>b.cost;
}
int main(){
    while(cin>>n){
        rep(i,1,n) p[i]=i;
        rep(i,1,n){
            a[i].ed=min((ll)n,read());
            a[i].cost=read();
        }
        sort(a+1,a+1+n,cmp);
        ll ans=0;
        rep(i,1,n) if(~find(a[i].ed)){
            ans+=a[i].cost;
        }
        println(ans);
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/caturra/p/8946629.html