本来想用这道题练习UFS,但是实在没有想出UFS的想法,于是写了个堆。贪心思路是:按时间降序贪心,每次放能放的最大的。代码如下:
#include<iostream>
using namespace std;
#include<cstdio>
#include<cstring>
#include<algorithm>
typedef short hd;
struct S{
hd p,d;
}a[10000];
inline bool cmp(S a,S b){
return a.d>b.d;
}
hd heap[10001],i,size;
inline void push(){
hd now=size++,next=now>>1;
heap[now]=a[i].p;
while(next){
if(heap[now]<=heap[next])return;
swap(heap[now],heap[next]);
now=next;
next=now>>1;
}
}
inline void pop(){
hd now=1,next=now<<1;
heap[1]=heap[--size];
while(next<size){
if(next<size&&heap[next]<heap[next+1])++next;
if(heap[now]>=heap[next])return;
swap(heap[now],heap[next]);
now=next;
next=now<<1;
}
}
int main(){
freopen("POJ1456.in","r",stdin);
hd n,j;
int sum;
while(scanf("%hd",&n)!=EOF){
for(i=0;i<n;++i)scanf("%hd%hd",&a[i].p,&a[i].d);
sort(a,a+n,cmp);
sum=0;
size=1;
for(j=10000,i=0;j>0;--j){
for(;i<n&&a[i].d==j;++i)
push();
if(size>1){
sum+=heap[1];
pop();
}
}
printf("%d\n",sum);
}
return 0;
}
用时:63MS
后来又看了UFS的题解,才发现UFS确实是更简洁的。思路是按权值降序贪心,每次填进能填的最靠后的时间。i的根节点表示截止到时间i为止最靠后的那段空闲时间,那么如果我们要占用这段时间,只需将其指向find(fnid(i)-1)即可,即指向这段空闲之前最靠后的那段空闲时间;如果find(i)=0,就说明i之前没有空闲时间了。
这实在是很巧妙的,它给我以两点启示:
1、以时间逆序选取最大权值,与,以权值逆序选取最后时间,是等价的。
2、并查集不仅可以表示与父亲的关系,也可以表示与根的关系。本题中,即是修改了这一想法,才得到了这一巧妙的做法。
UFS的代码是很简洁的,但时间上并没有太明显的优化,依然是63MS用时;但却是非常值得我学习的。
自己写的代码:
#include<iostream>
using namespace std;
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
typedef short hd;
struct S{
hd p,d;
}a[10000];
inline bool cmp(S a,S b){
return a.p>b.p;
}
hd fa[10001];
inline hd find(hd x){
return fa[x]!=fa[fa[x]]?fa[x]=find(fa[x]):fa[x];
}
int main(){
freopen("POJ1456.in","r",stdin);
hd n,i;
int sum;
while(scanf("%hd",&n)!=EOF){
for(i=0;i<n;++i)scanf("%hd%hd",&a[i].p,&a[i].d);
sort(a,a+n,cmp);
for(i=1;i<10001;++i)fa[i]=i;
sum=0;
for(i=0;i<n;++i)
if(find(a[i].d)){
sum+=a[i].p;
fa[fa[a[i].d]]=find(fa[a[i].d]-1);
}
printf("%d\n",sum);
}
return 0;
}