WA on test16 求调,3 个关注
查看原帖
WA on test16 求调,3 个关注
507348
__vector__楼主2023/2/6 14:44
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=998244353;
const int maxn=2e5+5;
int n,k;
int a[maxn],b[maxn];
ll dp[2][maxn];
// dp[0][i] 代表长度为 i 的 -1 段,两端不同  
// dp[1][i] 两端相同
ll ksm(ll a,ll b)
{
    ll ret=1;
    while(b)
    {
        if(b&1)ret=ret*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ret;
}
ll solve(int* a,int n)
{
    ll res=1;
    int i=1;
    while(i<=n)
    {
        if(a[i]!=-1)
        {
            if(a[i]==a[i-1])
            {
                res=0;
                break;
            }
            i++;
            continue;
        }
        else
        {
            int j=i;
            while(j<=n&&a[j+1]==-1)j++;
            if(i==1||j==n)
            {
                if(i==1&&j==n)
                {
                    res*=(ll)k;
                }
                else
                {
                    res*=ll(k-1);
                }
                res%=mod;
                res*=ksm(k-1,(j-i+1)-1);
                res%=mod;
            }
            else
            {
                if(a[i-1]!=a[j+1])
                {
                    res*=dp[0][j-i+1];
                }
                else
                {
                    res*=dp[1][j-i+1];
                }
                res%=mod;
            }
            i=j+1;
        }
    }
    return res;
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        int v;
        scanf("%d",&v);
        if(i&1)a[(i+1)>>1]=v;
        else b[i>>1]=v;
    }
    dp[0][0]=1;
    dp[0][1]=k-2;
    dp[1][1]=k-1;
    for(int i=2;i<=n;i++)
    {
        dp[0][i]=dp[1][i-1]*ll(k-1);
        dp[1][i]=dp[0][i-1]*ll(k-2)+dp[1][i-1];
        dp[0][i]%=mod;
        dp[1][i]%=mod;
    }
    printf("%lld",solve(a,(n>>1)+(n&1))*solve(b,n>>1)%mod);
    return 0;
}  
2023/2/6 14:44
加载中...