2019年牛客多校第三场 I(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;
}