CF 1622D Shuffle 组合数学 构造

题解:枚举最左和最右发生改变的位置,可看作将j-i+1个位置上的1重新分配。如果末端位置的数为0,直接分配一个1,如果末端位置的数为1,分配一个零,然后将C[c1+c0][c0]加给答案;
代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int mod=998244353;
int C[5005][5005],pre[5005];
int add(int x,int y){
    
    return (x+y)%mod;}
int main()
{
    
    
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    for(int i=0;i<=5000;i++)for(int j=0;j<=5000;j++)
    {
    
    
        if(i==0 || j==i)C[i][j]=1;
        else C[i][j]=add(C[i-1][j-1],C[i-1][j]);
    }
    int n,k,ans=1;
    cin>>n>>k;
    string t;
    cin>>t;
    for(int i=0;i<n;i++)pre[i+1]=pre[i]+t[i]-'0';
    for(int i=0;i<n;i++)
    {
    
    
        for(int j=i+1;j<n;j++)
        {
    
    
            int cnt=j-i+1;
            int cnt1=pre[j+1]-pre[i];
            if(pre[n]<k || cnt1>k)continue;
            cnt-=2;
            if(t[i]=='0')cnt1--;
            if(t[j]=='0')cnt1--;
            if(cnt>=cnt1 && cnt1>=0)
            {
    
    
                ans=add(ans,C[cnt][cnt1]);
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}

おすすめ

転載: blog.csdn.net/Ambrumf/article/details/122216715