2015 Multi-University Training Contest 10 hdu 5411 CRB and Puzzle

CRB and Puzzle

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 301    Accepted Submission(s): 127

 

Problem Description
CRB is now playing Jigsaw Puzzle.
There are N kinds of pieces with infinite supply.
He can assemble one piece to the right side of the previously assembled one.
For each kind of pieces, only restricted kinds can be assembled with.
How many different patterns he can assemble with at most M pieces? (Two patterns P and Q are considered different if their lengths are different or there exists an integer j such that j-th piece of P is different from corresponding piece of Q.)

Input
There are multiple test cases. The first line of input contains an integer T, indicating the number of test cases. For each test case:
The first line contains two integers N, M denoting the number of kinds of pieces and the maximum number of moves.
Then N lines follow. i-th line is described as following format.
$k a_1 a_2 ... a_k$
Here k is the number of kinds which can be assembled to the right of the i-th kind. Next k integers represent each of them.
$1 \leq T \leq 20$
$1 \leq N \leq 50$
$1 \leq M \leq 10^5$
$0 \leq k \leq N$
$1 \leq a1 < a2 < … < ak \leq N$

 

Output
For each test case, output a single integer - number of different patterns modulo 2015.

Sample Input
1
3 2
1 2
1 3
0

Sample Output
6

Hint
possible patterns are ∅, 1, 2, 3, 1→2, 2→3

Author
KUT（DPRK）

解题：矩阵快速幂+动态规划

$$\begin{bmatrix} ret & dp1 & dp2 & dp3 \\ \end{bmatrix} \times \begin{bmatrix} 1 & 0 & 0 & 0 \\ 1 & 0 & 0 & 0 \\ 1 & 1 & 0 & 0 \\ 1 & 0 & 1 & 0 \\ \end{bmatrix} $$

#include <bits/stdc++.h>
using namespace std;
const int maxn = 60;
const int mod = 2015;
int n,m;
struct Matrix {
    int m[maxn][maxn];
    void init() {
        memset(m,0,sizeof m);
    }
    void setOne() {
        init();
        for(int i = 0; i < maxn; ++i) m[i][i] = 1;
    }
    Matrix() {
        init();
    }
    Matrix operator*(const Matrix &t)const {
        Matrix ret;
        for(int k = 0; k <= n; ++k) {
            for(int i = 0; i <= n; ++i)
                for(int j = 0; j <= n; ++j)
                    ret.m[i][j] = (ret.m[i][j] + m[i][k]*t.m[k][j])%mod;
        }
        return ret;
    }
};
Matrix a,b,c;
int main() {
    int kase;
    scanf("%d",&kase);
    while(kase--) {
        scanf("%d%d",&n,&m);
        b.init();
        a.init();
        c.setOne();
        for(int i = 1,t,k; i <= n; ++i) {
            scanf("%d",&t);
            while(t--) {
                scanf("%d",&k);
                b.m[k][i] = 1;
            }
        }
        for(int i = 0; i <= n; ++i) {
            a.m[0][i] = 1;
            b.m[i][0] = 1;
        }
        a.m[0][0] = 0;
        while(m) {
            if(m&1) c = c*b;
            m >>= 1;
            b = b*b;
        }
        a = a*c;
        printf("%d\n",(a.m[0][0]+1)%mod);
    }
    return 0;
}