区间价值?hihocoder1483《二分!!+尺取法》

来源:互联网 发布:淘宝助手上传宝贝教程 编辑:程序博客网 时间:2024/05/29 13:57

区间价值?hihocoder1483《二分!!+尺取法》

给定n个数A1…An,小Ho想了解AL..AR中有多少对元素值相同。小Ho把这个数目定义为区间[L,R]的价值,用v[L,R]表示。
例如1 1 1 2 2这五个数所组成的区间的价值为4。
现在小Ho知道在所有的的v[L,R](1 <= L <= R <= n)中,第k小的值是多少。

Input
第一行一个数T(T<=10),表示数据组数。
对于每一组数据:
第一行两个数n,k(1<=n<=200,000,1<=k<=n*(n+1)/2)
第二行n个数A1…An(1<=Ai<=1,000,000,000)

Output
一个数表示答案。

Sample Input
2
4 7
1 1 2 3
3 6
100 100 100
Sample Output
0
3

题意很清晰了,想了很久尺取法,离散,但是不懂怎么存答案,区间太多,结果看大神的代码,竟然是二分答案,膜一波(别问谁是一波)。首先答案最大是n*(n+1)/2种(每个数都相等)。区间一共也是n*(n+1)/2种,于是我们将答案从0~n*(n+1)/2二分查询。假设查到当前答案为x,如果知道区间价值大于x的,就能求出价值小于x的,如果大于x的区间比n*(n+1)/2-k大,说明x比正确答案小。

#include<iostream>#include<cstdio>#include<string.h>#include<algorithm>using namespace std;const int N=200010;long long n,k;int a[N],sor[N],temp[N];//sor,temp为离散化过渡数组int num[N];//区间里i出现的次数bool getsum(long long cnt){    long long res=0,sum=0; //sum为满足价值大于cnt的区间数,res为当前区间的价值    int head=1,en=1;    memset(num,0,sizeof(num));    while(head<=n)    {        while(en<=n)        {            if(res+num[a[en]]<=cnt)            {                res+=num[a[en]]++;                en+=1;            }            else                break;        }        sum+=(n-en)+1;        num[a[head]]-=1;        res-=num[a[head]];        head+=1;    }    if(sum>(n+1)*n/2-k)        return true;    else        return false;}long long getans(long long left,long long right){    while(left<right)    {        long long m=(left+right)/2;        if(getsum(m)) left=m+1;        else            right=m;    }    return left;}int main(){    int T;    scanf("%d",&T);    sor[0]=0;    int tot;    while(T--)    {        tot=0;        scanf("%d%d",&n,&k);        for(int i=1;i<=n;i++)        {            scanf("%d",&a[i]);            sor[i]=a[i];        }        sort(sor+1,sor+n+1);        for(int i=1;i<=n;i++)   //离散        {            if(sor[i]==sor[i-1])                continue;            temp[tot++]=sor[i];        }        for(int i=1;i<=n;i++)        {            a[i]=lower_bound(temp+0,temp+tot-1,a[i])-temp;        }        long long ans=getans(0,n*(n+1)/2);        printf("%lld\n",ans);    }return 0;}