[COCI2017-2018 Contest4] Vode

一、题目

点此看题

二、解法

设$f[i][j]$为轮到第$i$个人，还剩$j$颗石子的胜负态，暴力转移$O(n^3)$，考虑前缀和优化。

设$l$为取最多石子能到达的数量，我们可以求得$[l,j-1]$的胜负态之和，现在讨论先后手的队伍：

• 如果先后手同属一队，那么和必须包含至少一个必胜态当前状态必胜，否则必败。
• 如果属于不同对，那么和只要包含至少一个必败态当前状态必胜，否则必败。

在$dp$时顺便维护一下前缀和即可，时间复杂度$O(n^2)$

#include <cstdio>
#include <iostream>
using namespace std;
const int M = 5005;
int read()
{
    int x=0,flag=1;
    char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*flag;
}
int n,m,k,a[M],f[M][M],s[M][M];
int main()
{
    n=read();m=read();k=read();
    for(int i=1;i<=n;i++)
        a[i]=read();
    for(int j=1;j<=m;j++)
        for(int i=1;i<=n;i++)
        {
            int t=i%n+1,l=max(1,j-k);
            int tmp=s[t][j-1]-s[t][l-1];
            if(a[i]==a[t])
            {
                if(tmp==0) f[i][j]=0;
                else f[i][j]=1;
            }
            else
            {
                if(tmp!=j-l) f[i][j]=1;
                else f[i][j]=0;
            }
            s[i][j]=s[i][j-1]+f[i][j];
        }
    for(int i=1;i<=n;i++)
        printf("%d ",!(f[i][m]^a[i]));
}