-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy path1034.cpp
More file actions
59 lines (51 loc) · 1.35 KB
/
1034.cpp
File metadata and controls
59 lines (51 loc) · 1.35 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
#include <iostream>
#include <map>
using namespace std;
struct Gang { int num, val, max; };
const int N = 2005;
bool marked[N];
int weights[N];
int graph[N][N];
map<string, Gang> ans;
map<string, int> name2id;
map<int, string> id2name;
int id(string name) {
if (name2id.count(name)) return name2id[name];
int new_id = name2id[name] = name2id.size();
id2name[new_id] = name;
return new_id;
}
void dfs(int v, Gang &gang) {
marked[v] = true;
gang.num++;
gang.val += weights[v];
gang.max = weights[v] > weights[gang.max] ? v : gang.max;
for (int w = 0; w < name2id.size(); w++)
if (!marked[w] && graph[v][w]) dfs(w, gang);
}
void dfs(int k) {
for (int v = 0; v < name2id.size(); v++) {
if (marked[v]) continue;
Gang gang{0, 0, v};
dfs(v, gang);
if (gang.num > 2 && gang.val > 2 * k)
ans[id2name[gang.max]] = gang;
}
}
int main() {
int n, k;
cin >> n >> k;
for (int i = 0; i < n; i++) {
int weight;
string name1, name2;
cin >> name1 >> name2 >> weight;
int v = id(name1), w = id(name2);
weights[v] += weight, weights[w] += weight;
graph[v][w] = graph[w][v] = weight;
}
dfs(k);
cout << ans.size() << endl;
for (auto pair : ans)
cout << pair.first << " " << pair.second.num << endl;
return 0;
}