(矩阵快速幂)hdu2157 How many ways??

题意大概是给出一个图的邻接矩阵,求出从a点走到b点长度恰好为k的路的条数。很基础的离散数学知识,这种路的条数即是邻接矩阵乘k次幂后对应位置a[i][j]的值,只不过数据范围较大,需要用矩阵快速幂来实现。其实是套模板稍加修改就可以过的一道题,但因为刚接触,天生愚钝,WA了无数次调了一个多小时才勉强ac,伤不起......

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int mod = 1000;
const int maxn = 20;
const int maxm = 100;

int n, m, T, A, B, k;

struct mat {
    ll a[maxn][maxn];
    mat() {
        memset(a, 0, sizeof(a));
    }
};

void print(mat ans) {
    for(int i = 0;i < n;i++) {
        for(int j = 0;j < n;j++)
            printf("%d ", ans.a[i][j]);
        printf("\n");
    }
    return;
}

mat init0() {
    mat tt;
    for(int i = 0;i < n;i++)
        for(int j = 0;j < n;j++) tt.a[i][j] = 0;
    return tt;
}

mat init1() {
    mat tt;
    for(int i = 0;i < n;i++)
        for(int j = 0;j < n;j++) tt.a[i][j] = (i==j);
    return tt;
}

mat mul(mat x, mat y) {
    mat tt;
    for(int i = 0;i < n;i++) {
        for(int j = 0;j < n;j++) {
            tt.a[i][j] = 0;
            for(int k = 0;k < n;k++) {
                tt.a[i][j] += x.a[i][k] * y.a[k][j];
            }
            tt.a[i][j] %= mod;
        }
    }
    return tt;
}

mat power(mat tt, int b) {
    mat res = init1();
    while(b) {
        if(b & 1) {
            res = mul(res, tt);
        }
        b >>= 1;
        tt = mul(tt, tt);
    }
    return res;
}

int main() {
    while(scanf("%d %d", &n, &m) != EOF && (n||m)) {
        int l, r;
        mat gra = init0();
        for(int i = 0;i < m;i++) {
            scanf("%d %d", &l, &r);
            gra.a[l][r] = 1;
        }
        scanf("%d", &T);
        while(T--) {
            scanf("%d %d %d", &A, &B, &k);
            mat b = power(gra, k);
            printf("%d\n", b.a[A][B]);
        }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值