中文题。
思路:起初一看以为是矩阵快速幂,因为求和和挑战程序设计竞赛那本书是一样的,后来发现是求不为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;
}