zoukankan      html  css  js  c++  java
  • HDU 6036

    /*
    HDU 6036 - Division Game [ 组合数学,NTT ]  |  2017 Multi-University Training Contest 1
    题意:
    	k堆石子围成一个圈,数量均为n,编号为0至k-1
    	第i轮可以操作第 (i+1) mod k 堆石子,必须拿石子且原石子数量要求整除操作后石子数量
    	任意一堆石子只剩一颗后停止游戏,问游戏停止在第i堆的方案数
    	限制:
    		n很大,按唯一分解定理形式给出 n = p1^e1 * p2^e2 * ... * pm^em
    		m,k <= 10	∑ei <= 1e5
    分析:
    	设 w = ∑ei ,则每堆石子最多操作w次
    	设 F(x) 为一个堆操作 x 次恰好变为1的方案数,则一个堆操作 x-1 次后不变为 1 的方案数也为 F(x)
    	对于石子堆i的方案数,枚举总操作次数x,设此时方案数为 ans[i][x]
    	则显然当石子堆i操作x次时,0 < j < i 的石子堆 j 均操作x次,而i < j <= k 的石子堆均操作x-1
    	故 ans[i][x] = [0<j<i的堆操作x次后不为1] * [堆i操作x次恰好为1] * [i<j<=k的堆操作x-1次后不为1]
    				 = F(x+1)^(i-1) * F(x) * F(x)^(k-i)
    				 = F(x+1)^(i-1) * F(x)^(k-i+1)
    	
    	现在研究一下上式 x 的取值范围
    		由于每个堆至多操作 w 次,则 [0<j<i的堆操作x次后不为1] 的 x ∈[0 , w-1]
    									[堆i操作x次恰好为1] 的 x ∈[0 , w]
    									[i<j<=k的堆操作x-1次后不为1] 的 x ∈[0 , w+1]
    		易得出结论 x ∈[0 , w-1]
    		但当 i = 1 时,0<j<i的堆 不存在,故其可以取到 x = w
    		所以分类讨论:
    			当 i = 1 时,ans[i][x] = F(x+1)^(i-1) * F(x)^(k-i+1) + F(w) , x ∈[0 , w-1]
    			当 i > 1 时,ans[i][x] = F(x+1)^(i-1) * F(x)^(k-i+1) , x ∈[0 , w-1]
    
    	研究如何求出 F(x):
    		此时有两个限制条件
    			1. 对于每个质因子pi,在 x 次内取完
    				即ei个相同的数字分成x组,允许空组,根据挡板法,方案数为 Comb(ei+x-1, x-1)
    				根据乘法原理 总方案数 F'(x) = ∏ Comb(ei+x-1, x-1) [1<=i<=m]
    				但由于存在第二个限制,F(x) != F'(x)
    			2. 每一次至少存在一个质因子被取
    				则根据容斥原理
    				F(x) = 随意取的方案数 - 某一次什么都没有取的方案数
    									  + 某两次什么都没有取的方案数
    									  - 某三次什么都没有取的方案数
    									  + ...
    				由于 x次中k次没有取的方案数 = Comb(x,k) * x-k次恰好取完的方案数 = Comb(x,k) *F(x-k)
    				则 F(x) = F'(x) - Comb(x,1) * F(x-1) 
    								+ Comb(x,2) * F(x-2) 
    								- Comb(x,3) * F(x-3)
    								+ ...
    						= Σ (-1)^i * C(x, i) * F(x-i) [0<=i<x]
    				将组合数打开优化:
    					F(x) = Σ (-1)^i * x! / i! / (x-i)! * F(x-i) [0<=i<x]
    					F(x)/x!  =  Σ (-1)^i/i!  *  F(x-i)/(x-i)!  [0<=i<x]
    				可以看出是卷积,再考虑模数特殊,用 NTT 优化
    
    对代码常数有一定要求
    */
    #include <bits/stdc++.h>
    using namespace std;
    #define LL long long
    const int N = 100005;
    const int MOD =  985661441;
    namespace NTT{
        const int G = 3;
        const int NUM = 20;
        int wn[20];
        int mul(int x, int y) {
            return (LL)x*y%MOD;
        }
        int PowMod(int a, int b) {
            int res = 1;
            a %= MOD;
            while (b) {
                if (b&1) res = mul(res, a);
                a = mul(a, a);
                b >>= 1;
            }
            return res;
        }
        void GetWn() {
            for (int i = 0; i < NUM; i++)
            {
                int t = 1<<i;
                wn[i] = PowMod(G, (MOD-1)/t);
            }
        }
        void Change(int a[], int len)
        {
            int i, j, k;
            for (i = 1, j = len/2; i < len-1; i++)
            {
                if (i < j) swap(a[i], a[j]);
                k = len/2;
                while (j >= k) {
                    j -= k;
                    k /= 2;
                }
                if (j < k) j += k;
            }
        }
        void NTT(int a[], int len, int on)
        {
            Change(a, len);
            int id = 0;
            for (int h = 2; h <= len; h <<= 1)
            {
                id++;
                for (int j = 0; j < len; j += h)
                {
                    int w = 1;
                    for (int k = j; k < j + h/2; k++)
                    {
                        int u = a[k] % MOD;
                        int t = mul(a[k+h/2], w);
                        a[k] = (u+t) % MOD;
                        a[k+h/2] = ((u-t)%MOD + MOD) % MOD;
                        w = mul(w, wn[id]);
                    }
                }
            }
            if (on == -1) {
                for (int i = 1; i < len/2; i++)
                    swap(a[i], a[len-i]);
                int inv = PowMod(len, MOD-2);
                for (int i = 0; i < len; i++)
                    a[i] = mul(a[i], inv);
            }
        }
    }
    int a[N<<3], b[N<<3];
    namespace COMB{
        int F[N<<1], Finv[N<<1], inv[N<<1];
        void init() {
            inv[1] = 1;
            for (int i = 2; i < N<<1; i++) {
                inv[i] = (LL)(MOD - MOD/i) * inv[MOD%i] % MOD;
            }
            F[0] = Finv[0] = 1;
            for (int i = 1; i < N<<1; i++) {
                F[i] = (LL)F[i-1] * i % MOD;
                Finv[i] = (LL)Finv[i-1] * inv[i] % MOD;
            }
        }
        int comb(int n, int m) {
            if (m < 0 || m > n) return 0;
            return (LL)F[n] * Finv[n-m] % MOD * Finv[m] % MOD;
        }
    }
    using namespace COMB;
    int e[20], m, k, n, tt;
    int g[N], ans[20];
    int pa[2][N];
    int main()
    {
        int i, x, len, tt = 0;
        NTT::GetWn();
        init();
        while (~scanf("%d%d", &m, &k))
        {
            n = 0;
            for (i = 1; i <= m; ++i)
            {
                scanf("%*d%d", &e[i]);
                n += e[i];
            }
            ++n;
            for (x = 0; x < n; ++x)
            {
                g[x] = 1;
                for (i = 1; i <= m; ++i)
                    g[x] = (LL)g[x] * comb(e[i]+x-1, x-1) % MOD;
            }
            for (i = 0; i < n; ++i)
            {
                a[i] = (i%2 ? MOD - Finv[i] : Finv[i]);
                b[i] = (LL)g[i] * Finv[i] % MOD;
            }
            len = 1;
            while (len < n*2) len <<= 1;
            for (i = n; i < len; ++i) a[i] = b[i] = 0;
            NTT::NTT(a, len, 1);
            NTT::NTT(b, len, 1);
            for (int i = 0; i < len; ++i) a[i] = NTT::mul(a[i], b[i]);
            NTT::NTT(a, len, -1);
            for (i = 0; i < n; ++i)
                a[i] = (LL)a[i] * F[i] % MOD;
            memset(ans, 0, sizeof(ans));
            a[n] = 0;
            int pre = 1, cur = 0;
            for (x = 1; x < n; ++x)
            {
                pre ^= 1, cur ^= 1;
                pa[cur][0] = 1;
                for (i = 1; i <= k; ++i)
                    pa[cur][i] = (LL)pa[cur][i-1] * a[x] % MOD;
                if (x > 1)
                    for (i = 1; i <= k; ++i)
                        ans[i] = (ans[i] + (LL)pa[cur][i-1] * pa[pre][k-i+1]) % MOD;
            }
            ans[1] += pa[cur][k];
            if (ans[1] > MOD) ans[1] -= MOD;
            printf("Case #%d:", ++tt);
            for (i = 1; i <= k; ++i) printf(" %d", ans[i]);
            puts("");
        }
    }
    

      

    *修正了一下写错的部分

    我自倾杯,君且随意
  • 相关阅读:
    记一次gogs迁徙
    Spark集群模式安装
    Spark单机模式安装
    SparkSQL入门
    SparkSql API
    Spark和HBase整合
    SparkStreaming与Kafka整合
    SparkStreaming基础案例
    Spark 自定义分区及区内二次排序demo
    Sqoop安装及指令
  • 原文地址:https://www.cnblogs.com/nicetomeetu/p/7259716.html
Copyright © 2011-2022 走看看