P8976「DTOI-4」排列 题解

一、题意

构造一个长度为 $n$ 的序列 $P$ ， 使 $P$ 的前半段的数之和 $\ge a$ ， 后半段的数之和 $\ge b$ 。

二、分析

1 $\sim n$ 每个数只能用一次，所以整个序列之和是 $\dfrac{n(n + 1)}{2} $（为方便，以下统称 $S1$ ） 半个序列之和最大是 $\frac{\frac{n}{2}\left(\frac{n}{2}+1+n\right)}{2}$ （为方便，以下统称 $S2$ ），如果 $a+b>S1$ 或 $a>S2$ 或 $b>S2$ ，就说明无解。

构造时我们只看前半个序列 （为方便，以下统称 $A$ ）。

一开始，我们将 $A$ 置为 $1\sim \frac{n}{2}$ ，和为$\frac{\frac{n}{2}\left(\frac{n}{2}+1\right)}{2}$ （为方便，以下统称 $S3$ ），如果 $S3\ge a$ ，那么就输出 $1\sim n$ 。

否则令 $T=a-S3$ ，$ X = T \div \dfrac{n}{2}$ ，$ F = T % \dfrac{n}{2}$ 。

下面来解释一下，若不够就得补上，每个数最多可以增大 $\frac{n}{2}$ ，我们可以将 $A$ 的后 $X$ 个数（ $\frac{n}{2}$ ，$\frac{n}{2}-1$ ，$\frac{n}{2}-2$ ，$\dots$ ）换成 $P$ 的后 $X$ 个数（ $n$ ，$n-1$ ，$n-2$ ，$\dots$ ），此时还会少 $F$ ，我们可以将 $A$ 的第 $\frac{n}{2}-X$ 个数换成 $P$ 的第 $\frac{n}{2}-X+F$ 个数（这一步在 $X\ne 0$ 的情况下才需要进行）。

这样序列 $A$ 就构造完成了，输出 $A$ ，然后将没用过的数输出即可。

三、代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll t,n,a,b;
int main()
{
    cin>>t;
    while(t--)
    {
    	cin>>n>>a>>b;
    	if(n*(n+1)/2<a+b||n*(n+1)/2-(n/2)*(n/2+1)/2<a||n*(n+1)/2-(n/2)*(n/2+1)/2<b)//判断是否无解 
    	{
    		cout<<-1<<endl;
    		continue;
		}
		ll t=(n/2)*(n/2+1)/2;//相当于上面的T 
		if(t<a)
		{
			ll t1=(a-t)/(n/2),x=(a-t)-t1*(n/2);//相当于上面的X、F 
			for(ll i=1;i<=n/2-t1-(bool)(x);i++)cout<<i<<" ";
			if(x)cout<<x+n/2-t1<<" ";
			for(ll i=n-t1+1;i<=n;i++)cout<<i<<" ";
			//构造A 
			for(ll i=n/2-t1;i<=n-t1;i++)
			{
				if(i==x+n/2-t1)continue;
				cout<<i<<" ";
			}
			//将没用过的数输出
			cout<<endl;
		}
		else
		{
			for(ll i=0;i<n;i++)cout<<i+1<<" ";
			cout<<endl;
		}
	}
    return 0;
}