sgu - 518 - Kidnapping（set + dp）

来源：互联网发布：android 无网络提示编辑：程序博客网时间：2024/05/20 14:17

题意：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;}