JZOJ 5440. 【NOIP2017提高A组冲刺11.1】背包

Description

蛤布斯有n种商品，第i种物品的价格为ai，价值为bi。有m个人来向蛤布斯购买商品，每个人每种物品只能购买一个。第j个人有cj的钱，他会不停选择一个能买得起的价格最高的商品买走（如果有多个则选择价值最高的）。你需要求出每个人购买的物品的价值和。

Input

第一行两个正整数n，m。接下来n行每行两个正整数ai，bi。接下来m行每行一个正整数cj。

Output

m行，每行一个整数表示答案。

Sample Input

5 4
10 5
9 8
7 3
3 4
1 2
20
100
28
18

Sample Output

15
22
18
10

Data Constraint

20%的数据，n,m<=1000。
30%的数据，ai,bi,cj在[1,10^12]中均匀随机。
100%的数据，n,m<=100000，ai,bi,cj<=10^12。

Solution

• 首先按价格从大到小排一遍序。

• 考虑跳着跳着走，每次先二分出能卖的最大价格的物品的位置，走过去、

• 之后二分出能卖的一段物品（用前缀和记录），减钱并购买之，跳到那一段的末尾。

• 之后再重复第一步，不断地一段一段走即可。

• 这样的时间复杂度为 $O(M\ log^2N)$ 。

• 因为一个人的钱数每次至少减少一半（价格从大到小，不然就还能再买），

• 这样就最多二分 $log$ 次，复杂度得证。

Code

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e5+2;
struct data
{
    long long x,y;
}a[N];
long long f[N],g[N];
inline long long read()
{
    long long X=0,w=1; char ch=0;
    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X*w;
}
inline void write(long long x)
{
    if(x>9) write(x/10);
    putchar(x%10+'0');
}
inline bool cmp(data x,data y)
{
    return x.x>y.x || x.x==y.x && x.y>y.y;
}
int main()
{
    int n=read(),m=read();
    for(int i=1;i<=n;i++) a[i].x=read(),a[i].y=read();
    sort(a+1,a+1+n,cmp);
    for(int i=1;i<=n;i++) f[i]=f[i-1]+a[i].x,g[i]=g[i-1]+a[i].y;
    while(m--)
    {
        long long c=read(),ans=0;
        if(c>=f[n])
        {
            write(g[n]),putchar('\n');
            continue;
        }
        for(int i=1;i<=n;i++)
        {
            int l=i,r=n,pos=0;
            if(a[i].x>c)
            {
                while(l<r)
                {
                    int mid=(l+r)>>1;
                    if(a[mid].x>c) l=mid+1; else r=mid;
                }
                if(a[l].x>c) break;
                i=l;
            }
            pos=l=i,r=n;
            while(l<r)
            {
                int mid=(l+r)>>1;
                if(f[mid]-f[i-1]<=c) l=mid+1,pos=mid; else r=mid;
            }
            c-=f[pos]-f[i-1];
            ans+=g[pos]-g[i-1];
            i=pos;
        }
        write(ans),putchar('\n');
    }
    return 0;
}