ARC153C ± Increasing Sequence

ARC153C ± Increasing Sequence

题目大意

给出一个长度为$n$的序列$A$，$A=(A_1,A_2,\dots ,A_n)$， A i ∈ { 1 , − 1 } A_i\in\{1,-1\} Ai​∈{1,−1}。

判断是否存在序列$x=(x_1,x_2\dots ,x_n)$，使得序列$x$满足以下条件：

• 对于所有的$1\le i\le n$，$|x_i|\le 10^{12}$
• $x$严格单调递增，即$x_1<x_2<\cdots <x_n$
• $\sum _{i=1}^n{A}_i{x}_i=0$

题解

如果直接处理$x$，因为$x$严格单调递增，所以会有些麻烦，我们不妨定义序列$y=(y_1,y_2\dots y_n)$，使得$y$满足以下条件：

$$y_i=\{\begin{array}{c}{x}_1,i=1\\ x_i-x_{i-1},i\ge 2\end{array}$$

那么$x_i=\sum _{k=1}^i{y}_k$，于是

$$\sum _{i=1}^n{A}_i{x}_i=\sum _{i=1}^n{A}_i\sum _{k=1}^i{y}_k=\sum _{k=1}^n(\sum _{i=k}^n{A}_i)y_k$$
令$B_k=\sum _{i=k}^n{a}_i$，$y$需满足以下条件：

• 对于所有$1\le i\le n$，$|\sum _{k=1}^i{y}_k|\le 10^{12}$
• 对于所有$2\le i\le n$，$y_i>0$
• $\sum _{i=1}^n{B}_i{y}_i=0$

这样，我们就将$x$严格单调递增的条件变成$y$的独立的条件。接下来，我们分情况考虑。

情况1：$B_1\ne 0$

若$B_1\ne 0$，我们可以构造以下一组解：

$$y_i=\{\begin{array}{c}-\sum _{i=2}^n{B}_i\times \frac{|B_1|}{B_1},i=1\\ \\ |B_1|,i\ge 2\end{array}$$

此时，$y$显然满足第二个条件和第三个条件。

对于第一个条件，因为

• 当$i=1$时，$|y_i|\le n(n-1)$
• 当$2\le i\le n$时，$|y_i|\le n$

所以$\sum _{i=1}^n|y_i|\le 2\times n^2\le 10^{12}$，满足第一个条件。

情况2：$B_1=0$

如果$B_i\ge 0$

如果$B_i\ge 0$，因为$B_1=0$，而当$2\le i\le n$时，$y_i>0$，所以
$$\sum _{i=1}^n{B}_i{y}_i=\sum _{i=2}^n{B}_i{y}_i>0$$

无法满足题意，所以无解。

如果$B_i\le 0$

如果$B_i\le 0$，与$B_i\ge 0$,无法满足题意，所以无解。

其他情况

因为存在$B_i$大于$0$，也存在$B_i$小于$0$，又因为$|B_{i+1}-B_i|=1$，所以一定存在$B_p=-1,B_q=1$。

令$X=\sum _{i=2}^n{B}_i$。如果$X\ge 0$，则令

$$y_i=\{\begin{array}{c}1,i\ne p\\ 1+Xi=p\end{array}$$

如果$X<0$，则令

$$y_i=\{\begin{array}{c}1,i\ne q\\ 1-Xi=q\end{array}$$

此时，$y$显然满足第二个条件和第三个条件。

对于第一个条件，因为$|B_i|\le n$，所以$|X|\le n(n-1)$，$\sum _{i=1}^n|y_i|\le |X|+n\le n^2\le 10^{12}$，满足第一个条件。

code

#include<bits/stdc++.h>
using namespace std;
int n,v1,v2,a[200005],b[200005];
long long x[200005],y[200005];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	for(int i=n;i>=1;i--){
		b[i]=b[i+1]+a[i];
		v1+=(b[i]>=0);
		v2+=(b[i]<=0);
	}
	if(b[1]!=0){
		printf("Yes\n");
		int f=1;
		if(b[1]<0) f=-1;
		for(int i=2;i<=n;i++){
			y[1]=y[1]-b[i]*f;
			y[i]=b[1]*f;
		}
		for(int i=1;i<=n;i++){
			x[i]=x[i-1]+y[i];
			printf("%lld ",x[i]);
		}
	}
	else{
		if(v1==n||v2==n) printf("No");
		else{
			printf("Yes\n");
			long long t=0;
			for(int i=2;i<=n;i++) t+=b[i];
			int f=0;
			if(t>=0){
				for(int i=1;i<=n;i++){
					if(b[i]==-1&&!f){
						y[i]=1+t;f=1;
					}
					else y[i]=1;
				}
			}
			else{
				for(int i=1;i<=n;i++){
					if(b[i]==1&&!f){
						y[i]=1-t;f=1;
					}
					else y[i]=1;
				}
			}
			for(int i=1;i<=n;i++){
				x[i]=x[i-1]+y[i];
				printf("%lld ",x[i]);
			}
		}
	}
	return 0;
}