2020-5 leetcode 1220. 统计元音字母序列的数目

1.用邻接矩阵创建一个有向图，两个节点之间有边等价于两个元音字母可以组合。
2.用离散数学中的一个理论，一个邻接矩阵的n次方 所形成的的矩阵中的每个元素代表 节点 i 到节点 j 所能形成的路径长度为n 的路径的个数。
3.一个长度为n 的路径代表一个 长度为n+1 的字符组合。
4.所以我们求出该邻接矩阵的 n - 1 次方，然后对矩阵元素求和就是答案。

class Solution {
public:
    const int mod=1e9+7;
    vector<vector<long long>> mul(vector<vector<long long>>& A,vector<vector<long long>>& B){
        int n=A.size(),m=B[0].size();
        vector<vector<long long>> nz(n,vector<long long>(m,0));
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                for(int k=0;k<A[i].size();k++){
                    nz[i][j]=(nz[i][j]+A[i][k]*B[k][j])%mod;
                }
            }
        }
        return nz;
    }
    vector<vector<long long>> pow(vector<vector<long long>> A,int n){
        int N=A.size();
        vector<vector<long long>> ans(N,vector<long long>(N,0));
        for(int i=0;i<N;i++) {
            ans[i][i]=1;
           
        }
        while(n>0){
            if(n&1==1){
                ans=mul(A,ans);
            }
            A=mul(A,A);
            n>>=1;
        }
        return ans;
    }
    int countVowelPermutation(int n) {
        vector<vector<long long>> grid={{0,1,0,0,0},{1,0,1,0,0},{1,1,0,1,1},{0,0,1,0,1},{1,0,0,0,0}};
        grid=pow(grid,n-1);
        vector<vector<long long>> ans={{1},{1},{1},{1},{1}};
        
        ans=mul(grid,ans);
        for(int i=1;i<5;i++){
            ans[i][0]+=ans[i-1][0]%mod;
        }
        return ans[4][0]%mod;
    }
};