首页 > 代码库 > hdu_4869_ 费马小引理+快速幂

hdu_4869_ 费马小引理+快速幂

Turn the pokers

  题意:

    给定m张牌,初始状态均为反面朝上。给定n次操作,每次指定所要翻的牌数,求n次操作后的牌的状态总数 mod 1000000009。

  输入:

    第一行n和m,代表n次操作和m张牌;

    第二行n个数,代表每次要翻的牌的张数。

  输出:

    方案数 mod 1000000009。

 

  思路:

    设反面为0,正面为1。对于一张牌翻两次和两张牌翻一次 得到的奇偶性相同,所以结果中最少的1(S)和最多的1(E)的奇偶性相同。如果找到了S和E,那么,介于这两个数之间且与这两个数奇偶性相同的数均可取到,然后在这个区间内组合数求解即可。

    因此这道题要解决好两个问题,一是求边界条件S和E,另一个是求在这之间内组合数求解。

    注意到这道题数据很大,C(n, m) = n!/(m!*(n-m)!),除法运算变得不合适。由 费马小定理 a^(p-1) = 1%p,那么,a^(p-2) = 1/a%p,利用这个公式,得到1/(m!*(n-m)!) = (m!*(n-m)!)^(p-2) mod p,即C(n, m) = n!*(m!*(n-m)!)^(p-2) mod p,这样就可以变除为乘。(也可以用求逆元 ing…)

    求(n-m)!)^p-2 mod p中用快速幂简化运算。

 

    代码:

  

 1 #include <cstdio> 2 #include <cstring> 3 #include <cmath> 4 #include <algorithm> 5 #define X 100005 6 using namespace std; 7  8 long long f[X]; 9 long long mod = 1000000009;10 11 void initial()12 {13     f[0] = 1;14     for(int i = 1; i < X; i++)15         f[i] = (f[i-1]*i)%mod;16 }17 18 // 快速幂19 long long PowerMod(long long a, long long b)20 {21     long long ans = 1;22     a = a % mod;23     while(b>0) {24         if(b % 2 == 1)25         ans = (ans * a) % mod;26         b = b/2;27         a = (a * a) % mod;28     }29     return ans;30 }31 32 int main()33 {34     int n, m, x;35     int s, e, S, E;36 37     initial(); // 求出n的阶乘38 39     while(scanf("%d%d", &n, &m) != EOF)40     {41         s = 0; e = 0;42         for(int i = 0; i < n; i++) {  // 找出最多和最少的1的个数43             scanf("%d", &x);44             if(x <= s) S = s - x;45             else if(e >= x) S = (s%2 == x%2)?0:1;46                 else S = x - e;47             if(e+x <= m) E = e + x;48             else if(s+x <= m) E = (((s+x)%2) == (m%2)?m:m-1);49                 else E = 2*m-(s+x);50             s = S;51             e = E;52         }53         long long ans = 0;54         for(int i = S; i <= E; i+=2) // 使用快速幂和费马小定理得出结果55             ans += ((f[m]%mod)*(PowerMod((f[i]*f[m-i])%mod, mod-2)%mod))%mod;56         printf("%lld\n", ans%mod);57     }58     return 0;59 }
View Code

 

    Ps:看了别人的代码,感觉都和别人的一样了,不熟练这样还可以理解,以后就再这样不行了。