题意:
给你一个数组,这个数组第i位是原数组第i,i+1,i+2位置上三个数的中位数
题解:
如果当前位置是i,第i-2个位置的中位数为x,第i-1个位置的中位数是y,第i个位置的中位数是z,那么ans[i]必为xyz中的某一个数,为什么?因为如果a[i]比任何一个数要大的话,那么它取最大值中位数也不会变,同理它如果比任意一个数小,取其中的最小值中位数也不会变。为了简化问题,我们就可以假设它一定在xyz这三个数之间。
那么设dp[i][j][k]表示在满足dp[i-2][k][l]的情况下,a[i]=v[i][j],a[i-1][=v[i-1][k],a[i-2]=v[i-2][l]成立。j,k,l都是0-2,v[i][j]表示这个位置的受影响的第j个中位数。
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
int dp[N][3][3],pre[N][3][3],val[N][3],a[N],ans[N];
int mid(int a,int b,int c)
{
if(a>b)
swap(a,b);
if(b>c)
swap(b,c);
if(a>b)
swap(a,b);
return b;
}
int main()
{
int t;
while(~scanf("%d",&t))
{
while(t--)
{
int n;
scanf("%d",&n);
for(int i=2;i<n;i++)
scanf("%d",&a[i]);
a[0]=a[1]=a[2],a[n+1]=a[n]=a[n-1];
for(int i=2;i<=n;i++)
for(int j=0;j<=2;j++)
for(int k=0;k<=2;k++)
dp[i][j][k]=0,dp[2][j][k]=1;
for(int i=1;i<=n;i++)
for(int j=0;j<=2;j++)
val[i][j]=a[i-1+j];
for(int i=3;i<=n;i++)
for(int j=0;j<=2;j++)
for(int k=0;k<=2;k++)
for(int l=0;l<=2;l++)
if(dp[i-1][k][l]&&mid(val[i][j],val[i-1][k],val[i-2][l])==a[i-1])
{dp[i][j][k]=1,pre[i][j][k]=l;break;}
int f1=0,f2;
for(int i=0;i<=2;i++)
for(int j=0;j<=2;j++)
if(dp[n][i][j])
{f1=i,f2=j;break;}
if(!f1)
{
printf("-1\n");
continue;
}
ans[n]=val[n][f1],ans[n-1]=val[n-1][f2];
int now=n-2;
int ne=pre[n][f1][f2];
while(now)
{
ans[now]=val[now][ne];
now--;
f1=f2;
f2=ne;
ne=pre[now+2][f1][f2];
}
for(int i=1;i<=n;i++)
printf("%d%c",ans[i],i==n?'\n':' ');
}
}
return 0;
}