-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtask12b.cpp
More file actions
68 lines (65 loc) · 1.92 KB
/
task12b.cpp
File metadata and controls
68 lines (65 loc) · 1.92 KB
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
#include <bits/stdc++.h>
using namespace std;
bool isLower(string s) {
for (int i = 0; i < s.size(); ++i) {
if (!islower(s[i])) {
return false;
}
}
return true;
}
void dfs(vector<vector<int>> &AL, vector<bool> visited, unordered_map<int, string> &umapRev, int curNode, int* numWays, bool visitTwice) {
if (umapRev[curNode] == "end") {
(*numWays)++;
return;
}
if (isLower(umapRev[curNode])) {
visited[curNode] = true;
}
for (auto &n: AL[curNode]) {
if (umapRev[n] != "start") {
if (!visited[n]) {
dfs(AL, visited, umapRev, n, numWays, visitTwice);
} else if (visited[n] && !visitTwice) {
dfs(AL, visited, umapRev, n, numWays, true);
}
}
}
}
int main() {
ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
string line;
vector<vector<int>> AL;
int curNode = 0;
unordered_map<string, int> umap;
unordered_map<int, string> umapRev;
while (getline(cin, line)) {
string node1, node2;
for (int i = 0; i < line.size(); ++i) {
if (line[i] == '-') {
node1 = line.substr(0, i);
node2 = line.substr(i+1);
}
}
if (umap.find(node1) == umap.end()) {
umap[node1] = curNode;
umapRev[curNode] = node1;
curNode++;
AL.push_back(vector<int>());
}
if (umap.find(node2) == umap.end()) {
umap[node2] = curNode;
umapRev[curNode] = node2;
curNode++;
AL.push_back(vector<int>());
}
AL[umap[node1]].push_back(umap[node2]);
AL[umap[node2]].push_back(umap[node1]);
}
int numWays = 0;
int start = umap["start"];
vector<bool> visited(curNode, false);
dfs(AL, visited, umapRev, start, &numWays, false);
cout << numWays;
return 0;
}