2019年牛客多校第三场 I Median（DP+思维）

这道题，主要还是要想通，如果某个数被三个中位数影响，那么这个数可以是其中一个。
证明：假设这个数，和三个中位数都不相等，那么这个数，要么大于最大的，要么小于最小的。举例（另一个例子就是对称的）：假设这个数大于最大的，如果想要改变某个中位数，那么便需要将这个最大的调小；如果想要不改变三个中位数，那么这个数不变（依旧最大），或者直接和对应中位数相等即可。即：这个数是一定可以等于其中一个的。然后DP即可。

DP：一个v[max][3],每一个v[i][3]中存储了能影响到当前a[i]的三个中位数，dp[i][j][k],这个用来村粗，a[i] = v[i][j] , a[i-1]= v[i-1][k],即：用来存储a[i]和a[i-1]对应了哪个中位数，最后dp[i][j][k]若是都对应上了，则其值为1，用于后面的递推。最后的pre[i][j][k]和DP数组很相似，是用来回溯写原串用的。

PS:最前面后最后的值可以直接等于相近的那个中位数，不会对答案有影响。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<vector>
#include<cmath>
#include<set>
#include<functional>
#include<algorithm>
#include<cassert>
#include<map>
using namespace std;
const int ma = 1e6;
int n;
long long b[ma];
long long v[ma][3];
long long dp[ma][3][3], pre[ma][3][3];
long long ans[ma];
int midd(int a,int b,int c) {
	int re[4];
	re[1] = a; re[2] = b; re[3] = c;
	sort(re + 1, re + 3 + 1);
	return re[2];
}
int main() {
	int t;
	scanf("%d", &t);
	while (t--) {
		scanf("%d", &n);
		for (int i = 3; i <= n; i++) {
			scanf("%lld", &b[i]);
		}
		b[2] = b[1] = b[3];
		b[n + 1] = b[n + 2] = b[n];
		v[1][0] = v[1][1] = v[1][2] = b[1];
		for (int i = 2; i <= n; i++) 
			v[i][0] = b[i], v[i][1] = b[i + 1], v[i][2] = b[i + 2];
		for (int i = 1; i <= n; i++)
			for (int j = 0; j <3; j++)
				for (int k = 0; k < 3; k++)
					dp[i][j][k] = pre[i][j][k] = 0;
		for (int i = 0; i < 3; i++)
			for (int j = 0; j <3; j++)
				dp[2][i][j] = 1;
		for (int i = 3; i <= n; i++) {
			for (int j = 0; j < 3; j++)
				for (int k = 0; k < 3; k++)
					for (int l = 0; l < 3; l++)
						if (dp[i - 1][k][l]) {
							if (midd(v[i][j], v[i - 1][k], v[i - 2][l]) == b[i])
							{
								dp[i][j][k] = 1;
								pre[i][j][k] = l;
								break;
							}
						}
		}
		int f = 0, jj, kk, ll;
		for (int j = 0; j < 3 && !f; j++) {
			for (int k = 0; k < 3; k++)
				if (dp[n][j][k]) {   //观察有没有一直到第N项了，那个数还是继承了中位数中的某个数
				                     //没有则说明，该串无对应的原串。
					jj = j;
					kk = k;
					f = 1;
					break;
				}
		}
		if (!f) { puts("-1\n"); continue; }
		for (int i = n; i >= 2; i--) {
			ans[i] = v[i][jj];
			ll = pre[i][jj][kk];
			jj = kk; kk = ll;
		}
		ans[1] = b[1];
		for (int i = 1; i <= n; i++) {
			printf("%lld",ans[i]);
			if (i != n) printf(" ");
		}
		printf("\n");
	}
	return 0;
}