cf1622 D. Shuffle

题意：

给定01串，你可以进行至多一次操作：选择一个恰含 K 个1的子段，任意改变这个子段中字符的顺序。问可能得到的不同串的数量

串长 5000

思路：

正常计数很容易就重复计算了。正解是暴力：

枚举第一个被改变（0变1、1变0）的位置 \(l\) 和最后一个被改变的位置 \(r\)（\(l<r\)）。注意只要 \([l,r]\) 中 1 的数量 \(\le K\) 就行。

\(l,r\) 一定要变，然后 \([l,r]\) 中还剩 \(ones\) 个 1 待分配，要把他们分配到 \((l,r)\) 上，也就是 \(C_{r-l-1}^{ones}\)

const signed N = 3 + 5000, mod = 998244353;
ll n, K, C[N][N], a[N], s[N]; //01串前缀和
signed main() {
    iofast;
    cin >> n >> K;

    for(int i = 0; i <= n; i++) { //initC
        C[i][0] = 1;
        for(int j = 1; j <= i; j++)
            C[i][j] = (C[i-1][j-1] + C[i-1][j]) % mod;
    }

    for(int i = 1; i <= n; i++) {
        char c; cin >> c;
        a[i] = c-'0', s[i] = s[i-1] + a[i];
    }

    if(s[n] < K) return cout << 1, 0; //特判1太少的情况

    ll ans = 1;
    for(int l = 1; l <= n; l++)
        for(int r = l + 1; r <= n; r++) {
            int ones = s[r] - s[l-1];
            if(ones > K) continue;
            ones -= (a[l] == 0) + (a[r] == 0);
            if(ones < 0) continue;
            ans += C[r-l-1][ones];
        }
    cout << ans % mod;
}

标签：ones,01,Shuffle,cout,5000,cf1622,signed,continue
来源： https://www.cnblogs.com/wushansinger/p/16275447.html