1211. 商人的宣传

1211. 商人的宣传
Constraints
Time Limit: 1 secs, Memory Limit: 32 MB
Description

 Bruce是K国的商人，他在A州成立了自己的公司，这次他的公司生产出了一批性能很好的产品，准备宣传活动开始后的第L天到达B州进行新品拍卖，期间Bruce打算将产品拿到各个州去做推销宣传,以增加其影响力。
K国有很多个州,每个州都与其他一些州相邻,但是K国对商人作宣传却有一些很奇怪的规定：
1、 商人只能从某些州到达另外一些州，即连通路线是单向的，而且有些州可能是到达不了的。
2、 商人不允许在同一个州连续宣传两天或以上，每天宣传完必须离开该州。
3、 商人可以多次来到同一个州进行宣传。
"我必须找出一条影响力最大的路线才行"，Bruce想，"但我首先必须知道到底有多少这种符合规定的宣传路线可供我选择。"现在Bruce把任务交给了你。并且出于考虑以后的需要，你还要帮他算出给出的两州之间的路线的总数。

Input
输入文件第一行包含三个整数n,m,L（1≤n,L≤100），分别表示K国的州数、连通路线的数量，以及多少天后必须到达B州。接下来有m行，每行一队整数x,y（1≤x,y≤n），表示商人能从x州到达y州。
第m+2行为一个整数q（1≤q≤100），表示Bruce有q个询问。下面q行每行两个整数A,B（1≤A,B≤n），即A、B州的位置。
Output
输出文件包含q行，每行一个整数t，为所求的从A州到B州满足上述规定的路线总数。
输入数据中的询问将保证答案t在长整数范围内，即t<231。
Sample Input
4 5 6
1 2
2 3
3 4
4 1
2 4
2
1 4
4 2
Sample Output
2
1

题目中：A->B->C花费两天时间，不用算A的时间，所有算出路径的长度就可以。看到题目先想到的是深度优先搜索，但是别人说会超时。然后别人教我了动态规划的思路：dp[I][j][l] = ∑dp[i'][j][l-1];即从I到j路径长为l的路线等于所有从i'到j的路径长为l-1的路线数目的和，其中i到i'的路径长为1，即i州可以到i'州。用vector数组储存n个州可以到达的下一个州。

#include <iostream>
#include <cstring>
#include <vector>
using namespace std;

const int MAX = 105;
int n, m, l, q;
vector<int> rodes[MAX];
int dp[MAX][MAX][MAX];

void findWays() {
    for (int k = 2; k <= l; k++) {
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= n; j++) {
                vector<int>::iterator it;
                for (it = rodes[i].begin(); it != rodes[i].end(); it++)
                    dp[i][j][k] += dp[*it][j][k - 1];
            }
        }
    }
}

int main() {
    int x, y;
    while (cin >> n >> m >> l) {
        memset(dp, 0, sizeof(dp));
    
        for (int i = 0; i < m; i++) {
            cin >> x >> y;
            dp[x][y][1] = 1;
            rodes[x].push_back(y);
        }

        findWays();

        cin >> q;
        int A, B;
        for (int i = 0; i < q; i++) {
            cin >> A >> B;
            cout << dp[A][B][l] << endl;
        }
        for(int i=1; i<=n; i++)//清空rodes
            rodes[i].clear();
    }
    return 0;
}                                 

过程中始终用l-1计算l,最后浪费了大量的数组空间，想出了另外一种方法，用两个数组二维数组存dp[i][j][l]和dp[i][j][l-1]即可。

#include <iostream>
#include <cstring>
#include <vector>
using namespace std;

const int MAX = 105;
int n, m, l, q;
vector<int> rodes[MAX];
int dp[MAX][MAX][2];

void findWays() {
	for (int k = 2; k <= l; k++) {//从路径长度为2开始计算
		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= n; j++) {
				vector<int>::iterator it;
				dp[i][j][k % 2] = 0;//每次计算前重置数组
				for (it = rodes[i].begin(); it != rodes[i].end(); it++)
					dp[i][j][k % 2] += dp[*it][j][(k - 1) % 2];//两个二维数组相互交替计算
			}
		}
	}
}

int main() {
	int x, y;
	while (cin >> n >> m >> l) {
		memset(dp, 0, sizeof(dp));
	
		for (int i = 0; i < m; i++) {
			cin >> x >> y;
			dp[x][y][1] = 1;//创建了路径长度为1的数组
			rodes[x].push_back(y);
		}

		findWays();

		cin >> q;
		int A, B;
		for (int i = 0; i < q; i++) {
			cin >> A >> B;
			cout << dp[A][B][l%2] << endl;
		}
		for(int i=1; i<=n; i++)
			rodes[i].clear();
	}
	return 0;
}