一、题目
二、解法
设 f [ i ] [ j ] f[i][j] f[i][j]为轮到第 i i i个人,还剩 j j j颗石子的胜负态,暴力转移 O ( n 3 ) O(n^3) O(n3),考虑前缀和优化。
设 l l l为取最多石子能到达的数量,我们可以求得 [ l , j − 1 ] [l,j-1] [l,j−1]的胜负态之和,现在讨论先后手的队伍:
- 如果先后手同属一队,那么和必须包含至少一个必胜态当前状态必胜,否则必败。
- 如果属于不同对,那么和只要包含至少一个必败态当前状态必胜,否则必败。
在 d p dp dp时顺便维护一下前缀和即可,时间复杂度 O ( n 2 ) O(n^2) O(n2)
#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]));
}