51nod1686 第K大区间

题目描述

1686 第K大区间
定义一个区间的值为其众数出现的次数。
现给出n个数，求将所有区间的值排序后，第K大的值为多少。

Input

第一行两个数n和k(1<=n<=100000,k<=n*(n-1)/2)
第二行n个数，0<=每个数<2^31

Output

一个数表示答案。

Input示例

4 2
1 2 3 2

Output示例

2

Solution

首先二分一个答案，然后再去判断答案是否可以。
如果直接暴力枚举区间是N^2的，所以要用一些奇技淫巧。首先枚举一个右端点，再找出使1-R一直到L-R这些区间里面值大于等于当前二分的mid的最大的L。那么，此时，对于这个右端点，贡献的答案就是L，累加一下统计就可以。
但是怎么维护这个最大的L呢？考虑当前的R向右移动，那么，L有可能往右移动也有可能不动，则对于A数组排序，找出一段相等的A[i],然后如果有长度大于等于mid的，就把右端点指向左端点，就可以维护出最大的L了

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long 
#define fo(i,a,b) for (int i=a;i<=b;i++)
struct edge{
   ll x,y;  
}  a[100050];
ll n,m,l,r,mid;
ll b[100050];
using namespace std;
bool cmp(edge a,edge b){
    return (a.x<b.x)||(a.x==b.x&&a.y<b.y);
}
int check(ll mid) {
    int fl=0;
    memset(b,0,sizeof(b));
    fo (i,mid,n) 
        if (a[i].x==a[i-mid+1].x) {
            fl=1;
            b[a[i].y]=a[i-mid+1].y;
        }
    if (!fl) return 0;
    ll ans=0;  ll l=0;
    fo (i,mid,n) {
       if (b[i]!=0) l=max(l,b[i]);
       ans+=l;
    }   
    if (ans>=m) return 1; else return 0;
}
int main(){
//freopen("1.in","r",stdin);
//  freopen("1.out","w",stdout);
    scanf("%lld%lld",&n,&m);
    fo(i,1,n) {
       scanf("%lld",&a[i].x);
       a[i].y=i;
}
    l=1; r=n;
    sort(a+1,a+n+1,cmp);
    while (l<=r) {
        mid=(l+r)>>1;
        if (check(mid)) l=mid+1; else r=mid-1;
    }
    printf("%lld",l-1);
}