【JZOJ 4932】 B

来源：互联网发布：理解大数据编辑：程序博客网时间：2024/04/30 05:53

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;}

0 0