SDUT 3911 指定长度路径数

Problem Description

题目给出一个有n个节点的有向图，求该有向图中长度为k的路径条数。方便起见，节点编号为1,2,…,n，用邻接矩阵表示该有向图。该有向图的节点数不少于2并且不超过500.

 

 

例如包含两个节点的有向图，图中有两条边1 → 2 ,2 → 1 。

长度为1的路径有两条：1 → 2 和 2 →1 ；

长度为2的路径有两条：1 → 2 → 1和2 → 1 → 2 ；

偷偷告诉你也无妨，其实这个图无论k取值多少 ( k > 0 )，长度为k的路径都是2条。

Input

多组输入，每组输入第一行是有向图中节点的数量即邻接矩阵的行列数n。接下来n行n列为该图的邻接矩阵。接下来一行是一个整数k.k小于30.

Output

输出一个整数，即为图中长度为k的路径的条数。

Sample Input

3
0 1 0
0 0 1
0 0 0
2

Sample Output

1

#include<iostream>

using namespace std;

int a[504][504];
int b[504][504];
int c[504][504];
int n;

void jzxc()
{
    for(int i = 0; i < n; i++)
        for(int j = 0; j < n; j++)
        {
            c[i][j] = 0;
            for(int k = 0; k < n; k++)
                c[i][j] += b[i][k]*a[k][j];
        }

        for(int i = 0; i < n; i++)
            for(int j = 0; j < n; j++)
                b[i][j] = c[i][j];
}

void show()
{
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < n; j++)
            cout<<b[i][j]<<" ";
        cout<<endl;
    }

}
int main()
{
    while(cin>>n)
    {
        for(int i = 0; i < n; i++)
            for(int j = 0; j < n; j++)
            {
                cin>>a[i][j];
                b[i][j] = a[i][j];
            }

        int k;
        cin>>k;
        while(--k) jzxc();

//        show();
        int ans = 0;
        for(int i = 0; i < n; i++)
            for(int j = 0; j < n; j++)
                ans += b[i][j];
        cout<<ans<<endl;
    }
}