-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsol.cpp
More file actions
executable file
·46 lines (41 loc) · 1.25 KB
/
Copy pathsol.cpp
File metadata and controls
executable file
·46 lines (41 loc) · 1.25 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
#include <bits/stdc++.h>
using namespace std;
int main() {
int countries, partners, home, leaver;
cin >> countries >> partners >> home >> leaver;
vector<vector<int>> adj(countries + 1);
vector<int> partnercount(countries + 1);
vector<int> currentcount(countries + 1);
vector<bool> handled(countries + 1);
queue<int> leaving;
leaving.push(leaver);
handled[leaver] = true;
int s, t;
for (int i = 0; i < partners; i++) {
cin >> s >> t;
adj[s].push_back(t);
adj[t].push_back(s);
partnercount[s]++;
currentcount[s]++;
partnercount[t]++;
currentcount[t]++;
}
int current;
while (!leaving.empty()) {
current = leaving.front();
leaving.pop();
for (int neighbour : adj[current]) {
currentcount[neighbour]--;
if (currentcount[neighbour] <= partnercount[neighbour] / 2 && handled[neighbour] == false) { // div by 2 is correct? check!
leaving.push(neighbour);
handled[neighbour] = true;
}
}
}
if (currentcount[home] <= partnercount[home] / 2 || home == leaver) {
cout << "leave" << endl;
} else {
cout << "stay" << endl;
}
return 0;
}