首页 诗词 字典 板报 句子 名言 友答 励志 学校 网站地图
当前位置: 首页 > 教程频道 > 开发语言 > 编程 >

hdu 4746 Bell 中国余下定理+矩阵乘法+第二类斯特林数

2013-10-08 
hdu 4746 Bell中国剩余定理+矩阵乘法+第二类斯特林数比赛的时候队友google到了,但没仔细想。google“Bell nu

hdu 4746 Bell 中国剩余定理+矩阵乘法+第二类斯特林数

比赛的时候队友google到了,但没仔细想。

google   “Bell number”  能得出以下结论:(注意p是质数)

http://en.wikipedia.org/wiki/Bell_number

hdu 4746 Bell  中国余下定理+矩阵乘法+第二类斯特林数

hdu 4746 Bell  中国余下定理+矩阵乘法+第二类斯特林数

{n k}是第二类斯特林数

题目中给的mod不是质数怎么办? 把它拆成多个质数(31, 37, 41, 43, 47)分别计算,然后用中国剩余定理合并一下。

n很大,我们先预处理出0--p的Bell数,然后矩阵乘法优化即可。

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;int Md[] = {31, 37, 41, 43, 47};int s[6][55][55], bell[6][55];void init() { //第二类斯特林数预处理    for(int i = 0;i < 5;i ++)        s[i][0][0] = 1;    for(int i = 1;i <= 50; i++) {        for(int j = 0;j < 5;j ++)            s[j][i][1] = 1;        for(int j = 1;j <= i; j++) {            for(int k = 0;k < 5; k++)                s[k][i][j] = (s[k][i-1][j-1] + j*s[k][i-1][j])%Md[k];        }        for(int j = 0;j < 5; j++) {            bell[j][i] = 0;            for(int k = 1;k <= i; k++)                bell[j][i] = (bell[j][i] + s[j][i][k])%Md[j];        }    }}int x[5];void mult(int a[55][55], int b[55][55], int n) { //矩阵乘法      int c[55][55] = {0};      int i, j, k;      for(k = 0; k < n; k++)            for(i = 0; i < n; i++) if(a[i][k])                  for(j = 0; j< n; j++)                        c[i][j] += a[i][k]*b[k][j];      for(i = 0; i < n; i++)            for(j = 0; j < n; j++)                  a[i][j] = c[i][j] % n;}int ans[55][55], A[55][55];int gao(int n, int mod, int id) {      if(n <= mod) return bell[id][n];      int i, j;      for(i = 0; i < mod; i++)            for(j = 0; j < mod; j++) {                  ans[i][j] = (i==j);                  A[i][j] = 0;            }      for(i = 0; i < mod-1; i++)            A[i+1][i] = 1;      A[0][mod-1] = A[1][mod-1] = 1;      n -= mod;      while(n > 0) {            if(n&1) mult(ans, A, mod);            mult(A, A, mod);            n >>= 1;      }      int ret = 0;      for(i = 0; i < mod; i++)            ret += bell[id][i+1]*ans[i][mod-1];      return ret%mod;}void ex_gcd(int a, int b, int &x, int &y) {      if(!b) {            x = 1; y = 0;      }      else { ex_gcd(b, a%b, y, x); y -= x*(a/b);}}int china(int n, int a[], int m[]) {    int M = 1;    for(int i = 0;i < n; i++) M *= m[i];    int ret = 0;    for(int i = 0;i < n; i++) {        int w = M/m[i], x, y;        ex_gcd(w, m[i], x, y);        ret = (ret + x*w*a[i])%M;    }    return (ret + M)%M;}int main() {      init();      int i, j, cas, n;      scanf("%d", &cas);      while(cas--) {            scanf("%d", &n);            for(i = 0; i < 5; i++)                  x[i] = gao(n, Md[i], i);            printf("%d\n", china(5, x, Md));      }      return 0;}


热点排行