【JZOJ 4932】 B

Description

现在你有N个数，分别为A1,A2,…,AN，现在有M组询问需要你回答。每个询问将会给你一个L和R(L<=R)，保证Max{Ai}-Min{Ai}<=R-L，你需要找出并输出最小的K(1<=K<=N，不存在输出-1)满足以下两个条件：
①能够在原来的N个数中选出不重复(下标不重复)的K个数，使得这K个数的和在区间[L,R]内。
②能够在原来的N个数中选出不重复(下标不重复)的K个数，使得这K个数的和不在区间[L,R]内。
N,M<=10^5，ai<=10^9

Analysis

显然A时无序的，排好序
条件②只要最大的k个加起来（设为mx[k]）大于R或最小的k个（设为mn[k]）加起来小于L即可
关键是①
根据②的思路，看看能不能根据mx[k]和mn[k]来确定答案
这是大体的思路方向，剩下的写起来较为麻烦，请读者自行思考

Code

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
using namespace std;
typedef long long ll;
const int N=100010;
int n,m,a[N];
ll mn[N],mx[N];
int find1(int l,int r,ll x)
{
    while(l<r)
    {
        int mid=(l+r)>>1;
        if(mx[mid]>=x) r=mid;
        else l=mid+1;
    }
    return l;
}
int find2(int l,int r,ll x)
{
    while(l<r)
    {
        int mid=(l+r)>>1;
        if(mx[mid]<=x) l=mid+1;
        else r=mid;
    }
    return l;
}
int main()
{
    ll x,y;
    scanf("%d %d",&n,&m);
    fo(i,1,n) scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    fo(i,1,n) mn[i]=mn[i-1]+a[i],mx[i]=mx[i-1]+a[n-i+1];
    mx[n+1]=2147483647;
    fo(i,1,m)
    {
        scanf("%lld %lld",&x,&y);
        int p=find1(1,n+1,x);
        if(p<=n && x>mn[p]) {printf("%d\n",p);continue;}
        else
        {
            p=find2(1,n+1,y);
            if(p<=n && mn[p]<=y) {printf("%d\n",p);continue;}
        }
        printf("-1\n");
    }
    return 0;
}