HDU 1759 Matrix Revolution(BFS)

中文题。

思路：起初一看以为是矩阵快速幂，因为求和和挑战程序设计竞赛那本书是一样的，后来发现是求不为0的个数。因为矩阵相乘的时候有 mat(i, j) = mat(i, k) *mat (k, j)的形式，这就要联想到图的矩阵表示，至少有一个k 使得(i,k) 和(k,j)都不为0，才能保证(i,j)的新值不为0，不妨设初始矩阵(i, j)位置如果为0，则表示i -> j有一条有向边，那么就是判断在K步以内从i是否能够到达j即可，bfs一下就行了。

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
const int maxn = 1010;
const int INF = 1e9;
using namespace std;

int d[maxn], tal;
int n, m;
vector<int> G[maxn];
char s[1010];

void init() {
    int len = strlen(s);
    if(len >= 9) tal = INF;
    else {
        tal = 0;
        for(int i = 0; i < len; i++)
            tal = tal * 10 + s[i] - '0';
    }
    for(int i = 0; i < n; i++) G[i].clear();
}

int sum(int x) {
    int ans = 0;
    for(int i = 0; i < n; i++) d[i] = INF;
    d[x] = 1;
    queue<int> q; q.push(x);
    while(!q.empty()) {
        int e = q.front(); q.pop();
        for(int i = 0; i < G[e].size(); i++) {
            int v = G[e][i];
            if(d[v] != INF) continue;
            d[v] = d[e] + 1;
            q.push(v);
        }
    }
    for(int i = 0; i < n; i++) {
        if(d[i] < tal) ans++;
    }
    return ans;
}

int main() {
    while(scanf("%d %d %s", &n, &m, s) != EOF) {
        init();
        int ans = 0;
        while(m--) {
            int f, t, tt;
            scanf("%d %d %d", &f, &t, &tt);
            G[f].push_back(t);
        }
        for(int i = 0; i < n; i++)
            ans += sum(i);
        printf("%d\n", ans);
    }
    return 0;
}