2019牛客暑期多校训练营（第三场）I Median —— DP通过中位数反推原数组

This way

题意：

给你一个数组，这个数组第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;
}