1. 程式人生 > 其它 >Solution -「ABC 215H」Cabbage Master

Solution -「ABC 215H」Cabbage Master

\(\mathcal{Description}\)

  Link.

  有 \(n\) 種顏色的,第 \(i\) 種有 \(a_i\) 個,任意兩球互不相同。還有 \(m\) 個盒子,每個盒子可以被放入某些顏色的小球,且第 \(i\) 個盒子要求放入總數不少於 \(b_i\)。你要拿走儘量少的球,使得要求無法被滿足,並求出此時拿球方案數模 \(998244353\) 的值。

  \(n\le20\)\(m\le10^4\)

\(\mathcal{Solution}\)

  如果保持清醒地做這道題還是比較簡單的。

  首先用 Hall 定理轉化合法條件,記 \(A=\{a_n\}\)\(B=\{b_m\}\)

\(\operatorname{adj}(S\subseteq B)\) 表示 \(S\) 內的 \(b\) 所鄰接的 \(a\) 的並,則合法條件為

\[\forall S\subseteq B,~\sum_{b\in S}b\le\sum_{a\in\operatorname{adj}(S)}a \]

那麼可以得知非法條件。固定 \(\operatorname{adj}(S)\),顯然 \(b\) 能多選就多選,最終能得到最小取球數量為

\[c=\max\left\{0,\min_{\operatorname{adj}(S)}\left\{\sum_{a\in\operatorname{adj}(S)}a-\sum_{\operatorname{adj}(\{b\})\subseteq\operatorname{adj}(S)}b+1\right\}\right\}. \]

  令 \(\mathcal S=\arg \min_{\operatorname{adj}(S)}\left\{\sum_{a\in\operatorname{adj}(S)}a-\sum_{\operatorname{adj}(\{b\})\subseteq\operatorname{adj}(S)}b+1\right\}\)

,也能避免算重地求出方案數為:

\[\sum_{T\in \mathcal S}\binom{\sum_{a\in T}a}{c}. \]

具體實現上,用幾次 FWT 即可。複雜度 \(\mathcal O(nm+2^nn)\)

\(\mathcal{Code}\)

/*~Rainybunny~*/

#include <cstdio>

#define rep( i, l, r ) for ( int i = l, rep##i = r; i <= rep##i; ++i )
#define per( i, r, l ) for ( int i = r, per##i = l; i >= per##i; --i )

inline void chkmin( int& a, const int b ) { b < a && ( a = b ); }

const int MAXN = 20, MAXM = 1e4, MAXV = 2e6, MOD = 998244353;
int n, m, a[MAXN + 5], b[MAXM + 5], adj[MAXM + 5], sum[1 << MAXN];
int fac[MAXV + 5], ifac[MAXV + 5], tot[1 << MAXN];
int cvr[1 << MAXN], chs[1 << MAXN];

inline void subeq( int& a, const int b ) { ( a -= b ) < 0 && ( a += MOD ); }
inline int sub( int a, const int b ) { return ( a -= b ) < 0 ? a + MOD : a; }
inline int mul( const long long a, const int b ) { return int( a * b % MOD ); }
inline int add( int a, const int b ) { return ( a += b ) < MOD ? a : a - MOD; }
inline void addeq( int& a, const int b ) { ( a += b ) >= MOD && ( a -= MOD ); }
inline int mpow( int a, int b ) {
    int ret = 1;
    for ( ; b; a = mul( a, a ), b >>= 1 ) ret = mul( ret, b & 1 ? a : 1 );
    return ret;
}

inline void init( const int s ) {
    fac[0] = 1;
    rep ( i, 1, s ) fac[i] = mul( i, fac[i - 1] );
    ifac[s] = mpow( fac[s], MOD - 2 );
    per ( i, s - 1, 0 ) ifac[i] = mul( ifac[i + 1], i + 1 );
}

inline int comb( const int a, const int b ) {
    return a < b ? 0 : mul( fac[a], mul( ifac[b], ifac[a - b] ) );
}

inline void fwtAND( const int len, int* u, const auto& adf ) {
    for ( int stp = 1; stp < len; stp <<= 1 ) {
        for ( int i = 0; i < len; i += stp << 1 ) {
            rep ( j, i, i + stp - 1 ) {
                adf( u[j], u[j + stp] );
            }
        }
    }
}

inline void fwtOR( const int len, int* u, const auto& adf ) {
    for ( int stp = 1; stp < len; stp <<= 1 ) {
        for ( int i = 0; i < len; i += stp << 1 ) {
            rep ( j, i, i + stp - 1 ) {
                adf( u[j + stp], u[j] );
            }
        }
    }
}

int main() {
    scanf( "%d %d", &n, &m );
    rep ( i, 0, n - 1 ) scanf( "%d", &a[i] );
    rep ( i, 0, m - 1 ) scanf( "%d", &b[i] );
    rep ( i, 0, n - 1 ) {
        rep ( j, 0, m - 1 ) {
            int t; scanf( "%d", &t );
            adj[j] |= t << i;
        }
    }
    rep ( i, 0, m - 1 ) sum[adj[i]] += b[i];
    fwtOR( 1 << n, sum, []( int& u, const int v ) { u += v; } );

    int tak = 1e9;
    rep ( S, 1, ( 1 << n ) - 1 ) {
        rep ( i, 0, n - 1 ) if ( S >> i & 1 ) tot[S] += a[i];
        if ( sum[S] ) chkmin( tak, tot[S] + 1 - sum[S] );
    }
    if ( tak <= 0 ) return puts( "0 1" ), 0;
    printf( "%d ", tak );

    int way = 0; init( tot[( 1 << n ) - 1] );
    rep ( S, 1, ( 1 << n ) - 1 ) {
        cvr[S] = tot[S] + 1 - sum[S] == tak;
        chs[S] = comb( tot[S], tak );
        if ( __builtin_popcount( S ) & 1 ) chs[S] = sub( 0, chs[S] );
    }
    fwtAND( 1 << n, cvr, []( int& u, const int v ) { u += v; } );
    fwtOR( 1 << n, chs, addeq );
    rep ( S, 1, ( 1 << n ) - 1 ) if ( cvr[S] ) {
        ( __builtin_popcount( S ) & 1 ? subeq : addeq )( way, chs[S] );
    }
    printf( "%d\n", way );
    return 0;
}

\(\mathcal{Details}\)

  某個問題無法找到解決方法時,尤其是在速度相關的比賽時,一定要冷靜下來,形式地描述“我想要求什麼東西”,而不是對著程式碼修修補補。