Div2 D. Range and Partition(构造,前缀和+二分枚举)

关于二分,从此以后我要改变写法了,我要使用一个ans值来记录最后一个正确的答案,因为这个题的确告诉我这样更稳妥一些。

D. Range and Partition

事实来讲,1800和1900之间是一道鸿沟,1800能做出来,1900直接把我卡的死死的

题意:给定一个长度为n的数组,请你将他分成k段,每一段都满足这一段在[x,y]区间里的数要比不再这个区间里的数多。

请你求出最小的[x,y]区间满足将数组分成k份,再请你将这k段的左右端点分k行输出

思路:难点不在求k段,肯定是在于求出合理的[x,y],那么我们不妨想一想怎么判断一个数组里在某个值域的数要多于不在某个值域的数。

显然,我们可以将所有值出现的次数都用pre数组记录,求出pre数组的前缀和,就可以获得落在某个区间内的数的个数。

那么合法的k个区间又怎么处理呢?

首先想到的是,如果隔板法,如果把不在[x,y]内的数视作隔板,那么显然当且仅当在[x,y]内的数的个数比不在[x,y]内的数的个数多k个时才有k个合法区间。

那么如果区间多了呢?

显然相邻的合法区间合并仍然合法,所以我们不用担心这个。

#include <iostream>

using namespace std;
#define int long long
#define endl '\n'
const int N = 2e5+100;
int a[N],pre[N];
int n,k;
void init()
{
    for(int i=1;i<=n;i++)
        pre[i]=0;
}
bool check(int l,int r)
{
    return pre[r]-pre[l-1]>=n-(pre[r]-pre[l-1])+k;
}
signed main()
{
    cin.tie(0);cout.tie(0);ios::sync_with_stdio(0);
    int t=1;
    cin>>t;
    while(t--)
    {
        cin>>n>>k;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
            pre[a[i]]++;
        }
        for(int i=1;i<=n;i++)
            pre[i]+=pre[i-1];
        int ansl=-1,ansr=n;
        for(int x=1;x<=n;x++)
        {
            int l=x,r=n,tempr=1e9;
            while(l<=r)
            {
                int mid=(l+r)/2;
                if(check(x,mid))
                {
                    tempr=min(tempr,mid);
                    r=mid-1;
                }
                else
                    l=mid+1;
            }
            if(tempr-x<ansr-ansl)
            {
                ansl=x;
                ansr=tempr;
            }
        }
        cout<<ansl<<" "<<ansr<<endl;
        int pre=1,in=0,out=0,num=1;
        for(int i=1;i<=n&&num<k;i++)
        {
            if(a[i]<=ansr&&a[i]>=ansl)
            {
                in++;
            }
            else
            {
                out++;
            }
            if(in>out)
            {
                cout<<pre<<" "<<i<<endl;
                num++;
                pre=i+1;
                in=0,out=0;
            }
        }
        cout<<pre<<" "<<n;
        if(t)
        {
            cout<<endl;
        }
        init();
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值