forked from ngthanhtrung23/CompetitiveProgramming
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathB.cpp
More file actions
92 lines (74 loc) · 2.32 KB
/
B.cpp
File metadata and controls
92 lines (74 loc) · 2.32 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include <bits/stdc++.h>
#define FOR(i,a,b) for(int i=(a),_b=(b); i<=_b; i++)
#define FORD(i,a,b) for(int i=(a),_b=(b); i>=_b; i--)
#define REP(i,a) for(int i=0,_a=(a); i<_a; i++)
#define EACH(it,a) for(__typeof(a.begin()) it = a.begin(); it != a.end(); ++it)
#define DEBUG(x) { cout << #x << " = "; cout << (x) << endl; }
#define PR(a,n) { cout << #a << " = "; FOR(_,1,n) cout << a[_] << ' '; cout << endl; }
#define PR0(a,n) { cout << #a << " = "; REP(_,n) cout << a[_] << ' '; cout << endl; }
#define sqr(x) ((x) * (x))
using namespace std;
int nPeople, d;
struct Person {
string name;
int k;
vector<string> children;
} people[1011];
vector<int> ke[1011];
int f[1011];
int dfs(int u, int cur, int need) {
if (cur+1 == need) return ke[u].size();
if (cur == need) return 1;
int res = 0;
for(int v : ke[u]) {
res += dfs(v, cur+1, need);
}
return res;
}
int main() {
ios :: sync_with_stdio(false);
int ntest; cin >> ntest;
FOR(test,1,ntest) {
cin >> nPeople >> d;
map<string,int> id;
FOR(i,1,nPeople) {
ke[i].clear();
people[i].children.clear();
cin >> people[i].name >> people[i].k;
id[people[i].name] = i;
REP(turn,people[i].k) {
string s; cin >> s;
people[i].children.push_back(s);
}
}
FOR(i,1,nPeople) {
int u = id[people[i].name];
for(auto s : people[i].children) {
int v;
if (id.count(s)) v = id[s];
else v = 0;
ke[u].push_back(v);
}
}
FOR(i,1,nPeople) {
f[i] = dfs(i, 0, d);
}
vector< pair<int, string> > res;
FOR(i,1,nPeople) if (f[i]) {
res.push_back(make_pair(-f[i], people[i].name));
}
sort(res.begin(), res.end());
int cnt = 0, last = 0;
cout << "Tree " << test << ":\n";
for(auto p : res) {
if (cnt < 3 || p.first == last) {
cout << p.second << ' ' << -p.first << '\n';
last = p.first;
++cnt;
}
else break;
}
cout << '\n';
}
return 0;
}