Algoogle

Algorithm for Programming Contest

PKU 2138 Travel Games

Category: PKU Tag: dfs

Travel Games

問題概要


辞書と初期の文字列が与えられる.
文字列に1文字追加した辞書上の単語に順に遷移できる.
たどり着くことができる辞書上の文字列のうち最長のものを求める.

解法


O(N^2)で遷移可能な文字列に辺を張ってDFSするだけ.

コード


(2138.cpp) download
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
#include <iostream>
#include <algorithm>
#include <queue>
#include <string>

using namespace std;

int N, s;
string st, ans;
string dict[1024];
vector<int> edge[1024];
bool visited[1024];

bool sim(int a, int b){
    if(dict[b].size() - dict[a].size() != 1) return 0;
    int match = 0;
    int j = 0;
    for(int i = 0; i < dict[b].size(); i++)
        if(dict[b][i] == dict[a][j]) j++;

    if(dict[a].size() == j) return 1;
    return 0;
}

void dfs(int id){
    if(visited[id]) return;
    visited[id] = 1;
    if(ans.size() < dict[id].size()) ans = dict[id];
    for(int i = 0; i < edge[id].size(); i++)
        dfs(edge[id][i]);
}

int main(){
    cin >> N >> st;
    for(int i = 0; i < N; i++){
        string str; cin >> str;
        dict[i] = str;
        if(str == st) s = i;
    }
    for(int i = 0; i < N; i++)
        for(int j = 0; j < N; j++)
            if(i != j && sim(i, j))
                edge[i].push_back(j);
    ans = st;
    dfs(s);
    cout << ans << endl;
    return 0;
}

Comments