POJ 1456 Supermarket(贪心+并查集优化)

一开始思路弄错了，刚开始想的时候误把所有截止时间为2的不一定一定要在2的时候买，而是可以在1的时候买。

举个例子：

50 2 10 1 20 2 10 1 50+20

50 2 40 4 30 4 20 1 10 1 20+50+30+40

思路：用优先级队列，每次取价格最大的(如果价格相同，取截止时间最大的)。

　　 然后往1~maxdx里加，首先看它截止时间上的位置是否已经存在其他物品，如果不存在，就加到该处。

　　 如果存在，就往前判断，直到有一处空位没被占用，就加入到该位置。

后来网上看了一下，可以用并查集查找不冲突的时间点

不用并查集优化，110ms；用并查集后，63ms

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;
const int maxn=10001;

int n;
int vis[maxn]; //若vis[i]=1，表示此处已经有物品，即有物品在i时刻卖出
int maxdx=0;  //用于记录所有物品中最大的截止时间
int f[maxn];
struct Product{
int price,deadtime;

bool operator <(const Product tmp) const{
if(price==tmp.price)
return deadtime<tmp.deadtime;
else
return price<tmp.price;
}
};

void init(){
for(int i=0;i<=maxdx;i++){
f[i]=i;
}
}
int find_root(int x){
if(f[x]!=x)
f[x]=find_root(f[x]);
return f[x];
}
int main()
{

int p,d;
int ans,counts;
Product tmp;
while(scanf("%d",&n)!=EOF){
priority_queue<Product> q;
maxdx=0;
memset(vis,0,sizeof(vis));
for(int i=1;i<=n;i++){
scanf("%d%d",&p,&d);
if(d>maxdx)
maxdx=d;
tmp.price=p;
tmp.deadtime=d;
q.push(tmp);

}
init();
ans=0;
counts=0;  //用于记录所卖的物品个数，如果个数为maxdx，表明1~maxdx中的所有时刻都已有物品在卖
while(counts<maxdx && !q.empty()){
tmp=q.top();
q.pop();
if(vis[tmp.deadtime]==0){
ans+=tmp.price;
vis[tmp.deadtime]=1;
f[tmp.deadtime]=tmp.deadtime-1;   //刚开始这里忘写了，导致WA
counts++;
}
else{
int t=tmp.deadtime;
int pos=find_root(t);
if(pos>0){
vis[pos]=1;
counts++;
ans+=tmp.price;
f[pos]=pos-1;
}
/*
while(vis[t]==1){
t--;
}
if(t>0){
vis[t]=1;
ans+=tmp.price;
counts++;
}
*/
}

}
printf("%d\n",ans);
}
return 0;
}