区间价值？hihocoder1483《二分！！+尺取法》

区间价值？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;
}