#30452 (C++11) No.92 逃走経路

• 問題: No.92 逃走経路
• ユーザー: ty70
• 提出日時: 2015-06-02 13:01:58
• 言語: C++11 (gcc 13.3.0)
• 結果: AC
• 実行時間: 194 ms / 5,000 ms
• コード長: 2,660 bytes
• コンパイル時間: 1,384 ms
• コンパイル使用メモリ: 104,568 KB
• 実行使用メモリ: 19,584 KB
• 最終ジャッジ日時: 2024-07-06 13:35:26
• 合計ジャッジ時間: 2,587 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

ファイルパターン	結果
sample	AC * 2
other	AC * 18

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#include <iostream>
#include <vector>
#include <string>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <algorithm>    // require sort next_permutation count __gcd reverse etc.
#include <cstdlib>  // require abs exit atof atoi 
#include <cstdio>       // require scanf printf
#include <functional>
#include <numeric>  // require accumulate
#include <cmath>        // require fabs
#include <climits>
#include <limits>
#include <cfloat>
#include <iomanip>  // require setw
#include <sstream>  // require stringstream 
#include <cstring>  // require memset
#include <cctype>       // require tolower, toupper
#include <fstream>  // require freopen
#include <ctime>        // require srand
#define rep(i,n) for(int i=0;i<(n);i++)
#define ALL(A) A.begin(), A.end()
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
/*
    No.92 逃走経路
    街 a, b 間の通行料金を c とすると
    multimap<int,pair<int,int> > cost で 通行料金 c とペアの街 (a,b), (b,a ) を保持。
    
    vector<P> cand[k] で 犯人が支払った通行料金と一致するペアの街 を保持。
    
    後は cand[i].second == cand[i+1].first となる経路を深さ優先探索で探索。
    結果を set<int> res に保持。
    計算量：O(M^K) やばい？ 1000^1000
    枝刈りしたら最悪ケースで大丈夫になった。
*/
const int MAX_N = 102;
const int MAX_K = 1005;
vector<P> cand[MAX_K];
bool visited[MAX_K][MAX_N];
set<int> res;
int N, M, K;
void dfs (int from, int depth ){
    if (depth == K ){
        res.insert (from );
        return;
    } // end if
    if (visited[depth][from] ) return;
    visited[depth][from] |= true;
    rep (j, cand[depth].size() ){
        P next = cand[depth][j];
        if (from == next.first ){
            dfs (next.second, depth + 1 );
        } // end if
    } // end rep
}
int main()
{
    memset (visited, false, sizeof (visited ) );
    ios_base::sync_with_stdio(0);
    cin >> N >> M >> K;
    multimap<int,P> cost; cost.clear();
    rep (i, M ){
        int a, b, c; cin >> a >> b >> c;
        cost.insert (make_pair(c, P (a, b ) ) );
        cost.insert (make_pair(c, P (b, a ) ) );
    } // end rep
    vector<int> d(K, 0 );
    rep (i, K ) cin >> d[i];
    rep (i, K ) cand[i].clear();
    multimap<int,P>::iterator it;
    rep (i, K ){
        int n = cost.count (d[i] );
        it = cost.find (d[i] );
        rep (j, n ){
            cand[i].push_back (it->second );
            it++;
        } // end rep
    } // end rep
    res.clear();
    rep (i, cand[0].size() ){
        int to = cand[0][i].second;
        dfs (to, 1 );
    } // end if
    cout << (int)res.size() << endl;
    set<int>::iterator jt = res.begin();
    for (; jt != res.end(); jt++ ){
        cout << (*jt) << ' ';
    } // end for
    if (!res.empty() )
        cout << endl;
    return 0;
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX