2021“MINIEYE杯”中国大学生算法设计超级联赛6 1005 Median（两种解法）

T1005 Median

这题蒟蒻会给出两种解法，一种是跟std一样的，另一种是基于贪心思想。

吐槽：特喵最后15min才发广播说这题题面出问题了，体验极其不好
题意：给定m个数字$a_i$，问能不能构造出一种分配方式，使得$1$到$n$的排列分成$m$个区间，第i个区间的中位数为$a_i$。

解法1：

1. 首先我们把被m个中位数分割开的数字分成一段一段的：假设 $n=6,m=2,b1=3,b2=5$，那么 $1,2,\cdot \cdot \cdot ,n$ 这些数会被 { 1 , 2 } \{ 1,2\} {1,2} { 4 } \{ 4\} {4} { 6 } \{ 6\} {6}三段，然后我们会发现任意两段中的任意一对数字一定可以配对消掉

2. 如果长度最大的段的数字个数小于等于其它段的数字个数之和，那么最终要么全部消掉，要么剩下一个，且剩下的这个数可以在任何一段内。如果会剩下，可以将最后一段的数字剩下最后一个（使得这个剩下的数数值最大），此时再把最后一段的数字放
   到中位数最小的集合中即可满足题意，所以答案一定为 YES。

3. 如果长度最大的段的数字个数大于其它段的数字个数之和，那么最终剩下的所有数字都在最大的这段内。设中位数小于这一段的最小值的集合的个数为 $x$，容易发现当且仅当 $x$ 不小于这一段剩下的数字个数时有解。

#include<bits/stdc++.h>
#define int long long
#define LL long long
#define N 101011
#define INF 0x3f3f3f3f
using namespace std;

int t,n,a[N],m,tot,maxx,ans,sum,l,r;

void solve()
{
	l = INF;
	r = 0;
	tot = 0;
	ans = 0;
	sum = 0;
	maxx = 0;
	int last = 1;
	scanf("%lld %lld",&n,&m);
	for(int i=1;i<=m;i++) scanf("%lld",&a[i]);
	if( m==0 )
	{
		printf("NO\n");
		return ;
	}
	sort(a+1,a+m+1);
	for(int i=1;i<=m;i++)
	{
		if( (a[i]-1 - last + 1)!=0 && maxx <= a[i]-1 - last + 1 )
		{
			l = last;
			r = a[i]-1;
			maxx = r - l + 1;
		}
		sum += a[i]-1 - last + 1;
		last = a[i]+1;
	}
	
	if( n-(a[m]+1)+1!=0 )
	{
		if( n-(a[m]+1)+1 >= maxx )
		{
			r = n;
			l = a[m]+1;
			maxx = r-l+1;
		}
		sum += n - (a[m]+1) + 1;
	}
	
	sum -= maxx;
	if( maxx <= sum )
	{
		printf("YES\n");
		return ;
	}
	int cnt = 0,yu = maxx-sum;
	for(int i=1;i<=m;i++)
	{
		if( a[i] <= l ) cnt++;
		else break;		
	}
	if( cnt>=yu )
	{
		printf("YES\n");
	}
	else printf("NO\n");
	return ;
}

signed main()
{
	//freopen("in.txt","r",stdin);
	cin>>t;
	while( t-- )
		solve();
	return 0;
}

解法2：

我们可以设$f_i$表示$b_i$后面最少要取几个数字，$g_i$表示$b_i$后面最多可以取几个数字，这样对于给定的$m$个中位数，我们都可以知道有哪些范围的$n$是符合规则的。

关于转移,首先初始化$f[1]=b[1]-1,g[1]=b[1]$，g数组部分很好转移，直接$g[i]=g[i-1]+b[i]-b[i-1]$。但是f数组我们需要讨论一下，首先设$len=b[i]-b[i-1]$

1. 当 $len>=f[i-1]$ 且 $len<=g[i-1]$ 时，$f[i]=0$
2. 当 $len>=f[i-1]$ 且 $len>g[i-1]$ 时，$f[i]=len-g[i-1]$
3. 当$len<f[i-1]$ 时， $f[i]=f[i-1]-len$；

最后注意一下特判，当 $m==0$ 时直接就不行，当 $m==1$ 时可行的$n$不再是一个范围而只能是一个数，即$(b_1+1)/2$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define LL long long
#define INF 0x3f3f3f3f
#define N 101010
using namespace std;

int n,m,b[N],f[N],g[N];

bool solve()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&b[i]);
	}
	if( m==0 ) return false;
	if( m==1 )
	{
		if( b[1] == ((n+1)/2) ) return true;
		else return false;
	}
	sort( b+1,b+m+1 );
	f[1] = b[1] - 1;
	g[1] = b[1];
	if( m==n )
	{
		return true;
	}
	int len;
	for(int i=2;i<=m;i++)
	{
		len = ( b[i] - b[i-1] - 1 );
		g[i] = g[i-1] + ( b[i] - b[i-1] ) ;
		if( len >= f[i-1] )
		{
			if( len <= g[i-1] ) f[i] = 0;
			else f[i] = len - g[i-1];
		}
		else f[i] = f[i-1] - len;
	}
	if( b[m]+f[m] > n || b[m]+g[m] < n ) return false;
	else return true;
}

int main()
{
	//freopen("in.txt","r",stdin);
	int t;
	cin>>t;
	while( t-- )
	{
		if( solve() ) printf("YES\n");
		else printf("NO\n");
	}
	
	return 0;
}