G - Supermarket

G - Supermarket

A supermarket has a set Prod of products on sale. It earns a profit px for each product x∈Prod sold by a deadline dx that is measured as an integral number of time units starting from the moment the sale begins. Each product takes precisely one unit of time for being sold. A selling schedule is an ordered subset of products Sell ≤ Prod such that the selling of each product x∈Sell, according to the ordering of Sell, completes before the deadline dx or just when dx expires. The profit of the selling schedule is Profit(Sell)=Σ  _x∈Sellpx. An optimal selling schedule is a schedule with a maximum profit. 
For example, consider the products Prod={a,b,c,d} with (pa,da)=(50,2), (pb,db)=(10,1), (pc,dc)=(20,2), and (pd,dd)=(30,1). The possible selling schedules are listed in table 1. For instance, the schedule Sell={d,a} shows that the selling of product d starts at time 0 and ends at time 1, while the selling of product a starts at time 1 and ends at time 2. Each of these products is sold by its deadline. Sell is the optimal schedule and its profit is 80. 

Write a program that reads sets of products from an input text file and computes the profit of an optimal selling schedule for each set of products. 

Input

A set of products starts with an integer 0 <= n <= 10000, which is the number of products in the set, and continues with n pairs pi di of integers, 1 <= pi <= 10000 and 1 <= di <= 10000, that designate the profit and the selling deadline of the i-th product. White spaces can occur freely in input. Input data terminate with an end of file and are guaranteed correct.

Output

For each set of products, the program prints on the standard output the profit of an optimal selling schedule for the set. Each result is printed from the beginning of a separate line.

Sample Input

4  50 2  10 1   20 2   30 1

7  20 1   2 1   10 3  100 2   8 2
   5 20  50 10

Sample Output

80
185

Hint

The sample input contains two product sets. The first set encodes the products from table 1. The second set is for 7 products. The profit of an optimal schedule for these products is 185.

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct node
{
    int pi;
    int di;
} w[10004];
bool cmp(node a,node b)
{
    if(a.pi!=b.pi)
        return a.pi>b.pi;
    else
        return a.di>b.di;
}
int main()
{
    int n,a[10004];
    while(~scanf("%d",&n))
    {
        int sum=0;
        memset(a,0,sizeof(a));
        for(int i=0; i<n; i++)
            scanf("%d%d",&w[i].pi,&w[i].di);
        sort(w,w+n,cmp);
        for(int i=0; i<n; i++)
        {
            while(w[i].di>0)//从截止日期往前搜索
            {
                if(a[w[i].di]==0)
                {
                    sum+=w[i].pi;
                    a[w[i].di]=1;
                    break;
                }
                w[i].di--;
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}

用并查集优化的代码：

#include<stdio.h>
#include<algorithm>
using namespace std;
int n,pre[10004];
struct node
{
    int pi;
    int di;
}w[10004];
bool cmp(node a,node b)
{
    if(a.pi!=b.pi)
        return a.pi>b.pi;
    return a.di>b.di;
}
int fond(int x)
{
    if(pre[x]==x)
        return x;
    return fond(pre[x]);
}
void init()
{
    for(int i=0;i<=10000;i++)
            pre[i]=i;
}
int main()
{
    while(~scanf("%d",&n))
    {
        init();
        for(int i=0;i<n;i++)
            scanf("%d%d",&w[i].pi,&w[i].di);
        sort(w,w+n,cmp);
        int ans=0;
        for(int i=0;i<n;i++)
        {
            int t=fond(w[i].di);
            if(t>0)
            {
                pre[t]--;
                ans+=w[i].pi;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}