sgu - 518 - Kidnapping（set + dp）

题意：n个点，给出这n个点的邻接矩阵，元素大小为对应边的长度，然后给出k个长度，问从结点1开始，依次走完这k个长度后最终到达的结点有哪几个（2 <= n <= 200, 1 <= k <= 200, 0 <= 矩阵元素 <= 200, 1 <= k个长度中每个长度 <= 200）。

题目链接：http://acm.sgu.ru/problem.php?contest=0&problem=518

——>>看n、k为200，不大，于是直接模拟一次，结果TLE去了。。。

接着想到了dp，设d[i]为依次走完前i个长度时能到达的点的集合（用set），则

状态转换方程为：d[i].insert(v[e]);（v[e]为d[i-1]中的元素经过第i个长度后能到达的点）。

#include <cstdio>
#include <vector>
#include <set>

using namespace std;

const int maxn = 200 + 10;
int head[maxn], nxt[maxn*maxn], u[maxn*maxn], v[maxn*maxn], w[maxn*maxn], r[maxn];
int n, k, ecnt;
set<int> d[maxn];

void init(){
    int i;
    for(i = 1; i <= n; i++) head[i] = -1;
    for(i = 1; i <= k; i++) d[i].clear();
    ecnt = 0;
}

void addEdge(int uu, int vv, int ww){
    u[ecnt] = uu;
    v[ecnt] = vv;
    w[ecnt] = ww;
    nxt[ecnt] = head[uu];
    head[uu] = ecnt;
    ecnt++;
}

void read(){
    int i, j, len;
    for(i = 1; i <= n; i++)
        for(j = 1; j <= n; j++){
            scanf("%d", &len);
            if(len) addEdge(i, j, len);
        }
    scanf("%d", &k);
    for(i = 1; i <= k; i++) scanf("%d", &r[i]);
}

void solve(){
    int i, e, u;
    set<int>::iterator p;
    for(e = head[1]; e != -1; e = nxt[e]){      //第一次后可到达哪
        if(w[e] == r[1]) d[1].insert(v[e]);
    }
    for(i = 2; i <= k; i++)     //dp
        for(p = d[i-1].begin(); p != d[i-1].end(); p++){
            u = *p;
            for(e = head[u]; e != -1; e = nxt[e]){
                if(w[e] == r[i]) d[i].insert(v[e]);
            }
        }
    if(d[k].empty()) printf("0\n");
    else{
        printf("%d\n", d[k].size());
        bool first = 1;
        for(p = d[k].begin(); p != d[k].end(); p++){
            if(first) first = 0;
            else printf(" ");
            printf("%d", *p);
        }
        printf("\n");
    }
}

int main()
{
    while(scanf("%d", &n) == 1){
        init();
        read();
        solve();
    }
    return 0;
}