D. Range and Partition

D. Range and Partition
题意:给你一个长为n的数组,让你分成k段,满足每段在[x,y]内的严格大于在在[x,y]外的,请你最小化y − x并输出切割方案。
思路:若有cnt1个数在[x,y]内 cnt2个数在[x,y]外 则整个区间一定能分成cnt1-cnt2段(类似中位数分段:传送门
cnt1+cnt2=n , cnt1-cnt2=k 则cnt1=(n+k+1)/2
可以用双指针求出 [x,y] 内有cnt1个数的最小区间

//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=8e18;
const int maxn=2e5+100;
int a[maxn],cnt[maxn];
signed main()
{
	IOS
	int tt;
	cin>>tt;
	while(tt--)
	{
		int n,k;
		cin>>n>>k;
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
			cnt[a[i]]++;
		}
		int cnt1=(n+k+1)/2,l=1,r=n,mn=n,now=0;
		for(int i=1,j=1;i<=n;i++)
		{
			while(j<=n&&now<cnt1) 
			{
				now+=cnt[j];
				j++;
			}
			if(now<cnt1)break;
			if(j-i<mn)
			{
				mn=j-i;
				l=i;
				r=j-1;
			}
			now-=cnt[i];
		}
		cout<<l<<" "<<r<<"\n";
		now=0;
		int i,j;
		for(i=1,j=1;i<=n;i=j)
		{
			if(k==1)break;
			now=0;
			while(j<=n&&now<1)
			{
				if(a[j]>=l&&a[j]<=r)now++;
				else now--;
				j++;
			}
			cout<<i<<" "<<j-1<<"\n";
			k--;
		}
		cout<<j<<" "<<n<<"\n";
		for(int i=1;i<=n;i++) cnt[i]=0;
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

thusloop

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值