1. 程式人生 > >POJ 1456 Supermarket(貪心演算法,可用並查集優化)

POJ 1456 Supermarket(貪心演算法,可用並查集優化)

題意:

        有n件商品需要賣,每件商品由(p,t)描述。其中p表示該商品被賣出可獲得的利潤,t表示該商品被賣出的截止時間。時間從1開始計時,每件商品被賣出的話需要佔用1個時間單位。如果某件商品的t=3,那麼該商品最多隻能在時間1,時間2或時間3 這3個時間點上賣。

        現在的問題是:對於給定的所有商品,我們如何安排每個時間點賣的商品能獲得最大利潤。輸出最大利潤即可。  

 #include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=10000+5;
 
struct Node//商品
{
    int p;//利潤
    int t;//截止時間
 
    bool operator<(const Node &rhs)const
    {
        return p>rhs.p;
    }
}nodes[maxn];
 
//findset(t)表示:t時間前的最晚一個可用時間點(可以是t)
int fa[maxn];
int findset(int x)
{
    return fa[x]==-1? x:fa[x]=findset(fa[x]);
}
 
int main()
{
    int n;
    while(scanf("%d",&n)==1)
    {
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&nodes[i].p,&nodes[i].t);
        }
        sort(nodes,nodes+n);
 
        memset(fa,-1,sizeof(fa));
        int cnt=0;//最終利潤
        for(int i=0;i<n;i++)
        {
            //找到i商品截止時間前的最晚一個可用時間點
            int time=findset( (nodes[i].t) );
            if(time>0)//時間可行
            {
                cnt += nodes[i].p;
                fa[time]=time-1;
            }
        }
 
        printf("%d\n",cnt);
    }
    return 0;
}