D. Range and Partition(贪心 + 双指针)

Problem - D - Codeforces

给定一个包含n个整数的数组a，找到一个值范围[æ， y] (z≤y)，并将a分割成恰好k(1≤k≤n)个子数组，如下所示:每个子数组由a的几个连续元素组成，即它等于a1, al+1，…， a，，对于某些l和r (1<l<r<n)。a中的每个元素只属于一个子数组。在每个子数组中，范围[æ， y](包括)内的元素数量严格大于范围外的元素数量。索引为i的元素在[æ， y]范围内当且仅当x <ai Sy。打印任意使y-æ最小的解。输入输入由多个测试用例组成。第一行包含一个整数t (1 < t < 3 - 104)——测试用例的数量。测试用例的描述如下。每个测试用例的第一行包含两个整数n和k (1 < k≤n≤2 - 105)——数组a的长度和分区所需的子数组的数量。每个测试用例的第二行包含n个整数a1,a2，， an (1 S an)，其中a是数组的第i个元素。它保证所有测试用例的n和不超过2- 105。输出对于每个测试用例，打印k+1行。在第一行中，打印z和y——找到的范围的极限。然后打印k行，第i行应该包含l和ri (1 <li <ri <n) -第i个子数组的极限。您可以以任何顺序打印子数组。

Example

input

Copy

3

2 1

1 2

4 2

1 2 2 2

11 3

5 5 5 1 5 5 1 5 5 5 1

output

Copy

1 2
1 2
2 2
1 3
4 4
5 5
1 1
2 2
3 11

请注意在第一个测试中，应该只有一个子数组，它必须等于整个数组。在[1,2]范围内有2个元素，在[1,2]范围外有O个元素，如果选择的范围为[1,1)，则在(a1)范围内有1个元素，在(a2)范围外有1个元素，则答案无效。在第二个测试中，可以选择范围[2,2]，将数组拆分为子数组(1,3)和子数组(4,4)，在子数组(1,3)中有2个元素在范围内(a2和ag)和1个元素在范围外(a1)，在子数组(4,4)中只有1个元素(a4)，并且在范围内。在第三个测试中，可以选择范围[5,5)，并将数组拆分为子数组(1,4)，(5,7)和(8,11)，在子数组(1,4)中有3个元素在范围内和1个元素在外部，在子数组(5,7)中有2个元素在内部和1个元素在外部，在子数组(8,11)中有3个元素在内部和1个元素在外部。

题解:
假设cnt1为f(l,r)区间在(x,y)范围内的,cnt2为不在的,

f(l,mid) + f(mid+1,r) =  f(l,r) 如果f(l,r) > 0

那么f(l,r) == k是不是也可以由一个个小的区间f(l,r) == 1的区间合并过来呢

由于我们是求一个范围,我们先把数组里的数映射到一个数组上,记录哪个数出现几次,

如果要符合题意,应满足

cnt1 - cnt2 == k

cnt1 + cnt2 = n

cnt1 = (n + k) / 2

所以我们应该利用双指针,找到一个最小的,包含这么多数的区间,

找的区间后,输出即可

然后遍历数组,每次区间够1输出即可

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<vector>
#include<map>
#include<queue>
#include<set>
#include<cstdio>
using namespace std;
//#define int long long
const int N = 2e5 + 10;
typedef pair<int, int> PII;
typedef long long ll;
int a[N],n,k,s[N];
void solve() 
{
	cin >> n >> k;
	for(int i = 1;i <= n;i++)
	{
		scanf("%d",&a[i]);
		s[a[i]]++;
	}
	int len = (k + n + 1)/2;
	int l = 1,r = n;
	int sum = 0;
	int mi = n;
	for(int i = 1,j = 1;i <= n;i++)
	{
		while(j <= n&&sum < len)
		{
			sum += s[j++];
		}
		if(sum < len)
		break;
		if(j - i < mi)
		{
			mi = j - i;
			r = j - 1;
			l = i;
		}
		sum -= s[i];
	}
	printf("%d %d\n",l,r);
	int last = 1;
	sum = 0;
	for(int i = 1;i <= n;i++)
	{
		if(a[i] >= l&&a[i] <= r)
		{
			sum ++;
		}
		else
		{
			sum--;
		}
		if(sum > 0)
		{
			if(k != 1)
			{
				printf("%d %d\n",last,i); 
			}
			else
			{
				printf("%d %d\n",last,n);
				break;
			}
			k --;
			sum = 0;
			last = i + 1;
		
		}
	}
	for(int i = 1;i <= n;i++)
	{
		s[i] = 0;
	}
}

//1 2 4
signed main() 
{
//	ios::sync_with_stdio(0);
//	cin.tie(0);cout.tie(0);
	int t = 1;
//	cin >> t;
scanf("%d",&t);
	while (t--) 
	{
		solve();
	}
}
//1 1 1 0 1

//1 1 1 0 1
//1 1 1 0 1
//1 1 1 0 1
//0 1 1 1 1
//0 1 1 1 1