Submission #606824

# Submission time Handle Problem Language Result Execution time Memory
606824 2022-07-26T08:47:05 Z Hanksburger Railway (BOI17_railway) C++17
100 / 100
219 ms 34212 KB
#include <bits/stdc++.h>
using namespace std;
vector<pair<int, int> > adj[100005];
map<int, int> mp[100005];
int s[100005], n, m, k;
vector<int> ans;
void dfs(int u, int p, int ind)
{
    for (int i=0; i<adj[u].size(); i++)
    {
        int v=adj[u][i].first, w=adj[u][i].second;
        if (v!=p)
        {
            dfs(v, u, w);
            if (mp[u].size()<mp[v].size())
                mp[u].swap(mp[v]);
            for (auto itr=mp[v].begin(); itr!=mp[v].end(); itr++)
            {
                mp[u][itr->first]+=itr->second;
                if (mp[u][itr->first]==s[itr->first])
                    mp[u].erase(itr->first);
            }
        }
    }
    if (mp[u].size()>=k)
        ans.push_back(ind);
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m >> k;
    for (int i=1; i<n; i++)
    {
        int u, v;
        cin >> u >> v;
        adj[u].push_back({v, i});
        adj[v].push_back({u, i});
    }
    for (int i=1; i<=m; i++)
    {
        cin >> s[i];
        for (int j=1; j<=s[i]; j++)
        {
            int u;
            cin >> u;
            mp[u][i]++;
        }
    }
    dfs(1, 1, 0);
    cout << ans.size() << '\n';
    sort(ans.begin(), ans.end());
    for (int i:ans)
        cout << i << ' ';
    return 0;
}

Compilation message

railway.cpp: In function 'void dfs(int, int, int)':
railway.cpp:9:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    9 |     for (int i=0; i<adj[u].size(); i++)
      |                   ~^~~~~~~~~~~~~~
railway.cpp:25:21: warning: comparison of integer expressions of different signedness: 'std::map<int, int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   25 |     if (mp[u].size()>=k)
      |         ~~~~~~~~~~~~^~~
# Verdict Execution time Memory Grader output
1 Correct 4 ms 7252 KB Output is correct
2 Correct 7 ms 7908 KB Output is correct
3 Correct 7 ms 7900 KB Output is correct
4 Correct 4 ms 7372 KB Output is correct
5 Correct 4 ms 7368 KB Output is correct
6 Correct 8 ms 8532 KB Output is correct
7 Correct 8 ms 8020 KB Output is correct
8 Correct 8 ms 7892 KB Output is correct
9 Correct 9 ms 8020 KB Output is correct
10 Correct 4 ms 7380 KB Output is correct
11 Correct 4 ms 7252 KB Output is correct
12 Correct 4 ms 7252 KB Output is correct
13 Correct 3 ms 7380 KB Output is correct
14 Correct 4 ms 7252 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 4 ms 7252 KB Output is correct
2 Correct 7 ms 7908 KB Output is correct
3 Correct 7 ms 7900 KB Output is correct
4 Correct 4 ms 7372 KB Output is correct
5 Correct 4 ms 7368 KB Output is correct
6 Correct 8 ms 8532 KB Output is correct
7 Correct 8 ms 8020 KB Output is correct
8 Correct 8 ms 7892 KB Output is correct
9 Correct 9 ms 8020 KB Output is correct
10 Correct 4 ms 7380 KB Output is correct
11 Correct 4 ms 7252 KB Output is correct
12 Correct 4 ms 7252 KB Output is correct
13 Correct 3 ms 7380 KB Output is correct
14 Correct 4 ms 7252 KB Output is correct
15 Correct 50 ms 13616 KB Output is correct
16 Correct 40 ms 13676 KB Output is correct
17 Correct 46 ms 13772 KB Output is correct
18 Correct 12 ms 8668 KB Output is correct
19 Correct 8 ms 8020 KB Output is correct
20 Correct 50 ms 13080 KB Output is correct
21 Correct 48 ms 13052 KB Output is correct
22 Correct 4 ms 7252 KB Output is correct
23 Correct 7 ms 7988 KB Output is correct
24 Correct 8 ms 7892 KB Output is correct
25 Correct 4 ms 7288 KB Output is correct
26 Correct 4 ms 7368 KB Output is correct
27 Correct 9 ms 8548 KB Output is correct
28 Correct 8 ms 7952 KB Output is correct
29 Correct 7 ms 7924 KB Output is correct
30 Correct 8 ms 8020 KB Output is correct
31 Correct 3 ms 7380 KB Output is correct
32 Correct 4 ms 7252 KB Output is correct
33 Correct 3 ms 7252 KB Output is correct
34 Correct 3 ms 7380 KB Output is correct
35 Correct 4 ms 7368 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 125 ms 29996 KB Output is correct
2 Correct 4 ms 7380 KB Output is correct
3 Correct 116 ms 29748 KB Output is correct
4 Correct 110 ms 29268 KB Output is correct
5 Correct 85 ms 29152 KB Output is correct
6 Correct 103 ms 31328 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 107 ms 24868 KB Output is correct
2 Correct 136 ms 20840 KB Output is correct
3 Correct 158 ms 26524 KB Output is correct
4 Correct 138 ms 26832 KB Output is correct
5 Correct 151 ms 26644 KB Output is correct
6 Correct 130 ms 25144 KB Output is correct
7 Correct 125 ms 25076 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 107 ms 24868 KB Output is correct
2 Correct 136 ms 20840 KB Output is correct
3 Correct 158 ms 26524 KB Output is correct
4 Correct 138 ms 26832 KB Output is correct
5 Correct 151 ms 26644 KB Output is correct
6 Correct 130 ms 25144 KB Output is correct
7 Correct 125 ms 25076 KB Output is correct
8 Correct 96 ms 23884 KB Output is correct
9 Correct 92 ms 25292 KB Output is correct
10 Correct 72 ms 30520 KB Output is correct
11 Correct 72 ms 30412 KB Output is correct
12 Correct 61 ms 17336 KB Output is correct
13 Correct 63 ms 17380 KB Output is correct
14 Correct 114 ms 23280 KB Output is correct
15 Correct 139 ms 23628 KB Output is correct
16 Correct 177 ms 29504 KB Output is correct
17 Correct 161 ms 29940 KB Output is correct
18 Correct 150 ms 27884 KB Output is correct
19 Correct 149 ms 22916 KB Output is correct
20 Correct 115 ms 26840 KB Output is correct
21 Correct 110 ms 26964 KB Output is correct
22 Correct 115 ms 27008 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 4 ms 7252 KB Output is correct
2 Correct 7 ms 7908 KB Output is correct
3 Correct 7 ms 7900 KB Output is correct
4 Correct 4 ms 7372 KB Output is correct
5 Correct 4 ms 7368 KB Output is correct
6 Correct 8 ms 8532 KB Output is correct
7 Correct 8 ms 8020 KB Output is correct
8 Correct 8 ms 7892 KB Output is correct
9 Correct 9 ms 8020 KB Output is correct
10 Correct 4 ms 7380 KB Output is correct
11 Correct 4 ms 7252 KB Output is correct
12 Correct 4 ms 7252 KB Output is correct
13 Correct 3 ms 7380 KB Output is correct
14 Correct 4 ms 7252 KB Output is correct
15 Correct 50 ms 13616 KB Output is correct
16 Correct 40 ms 13676 KB Output is correct
17 Correct 46 ms 13772 KB Output is correct
18 Correct 12 ms 8668 KB Output is correct
19 Correct 8 ms 8020 KB Output is correct
20 Correct 50 ms 13080 KB Output is correct
21 Correct 48 ms 13052 KB Output is correct
22 Correct 4 ms 7252 KB Output is correct
23 Correct 7 ms 7988 KB Output is correct
24 Correct 8 ms 7892 KB Output is correct
25 Correct 4 ms 7288 KB Output is correct
26 Correct 4 ms 7368 KB Output is correct
27 Correct 9 ms 8548 KB Output is correct
28 Correct 8 ms 7952 KB Output is correct
29 Correct 7 ms 7924 KB Output is correct
30 Correct 8 ms 8020 KB Output is correct
31 Correct 3 ms 7380 KB Output is correct
32 Correct 4 ms 7252 KB Output is correct
33 Correct 3 ms 7252 KB Output is correct
34 Correct 3 ms 7380 KB Output is correct
35 Correct 4 ms 7368 KB Output is correct
36 Correct 125 ms 29996 KB Output is correct
37 Correct 4 ms 7380 KB Output is correct
38 Correct 116 ms 29748 KB Output is correct
39 Correct 110 ms 29268 KB Output is correct
40 Correct 85 ms 29152 KB Output is correct
41 Correct 103 ms 31328 KB Output is correct
42 Correct 107 ms 24868 KB Output is correct
43 Correct 136 ms 20840 KB Output is correct
44 Correct 158 ms 26524 KB Output is correct
45 Correct 138 ms 26832 KB Output is correct
46 Correct 151 ms 26644 KB Output is correct
47 Correct 130 ms 25144 KB Output is correct
48 Correct 125 ms 25076 KB Output is correct
49 Correct 96 ms 23884 KB Output is correct
50 Correct 92 ms 25292 KB Output is correct
51 Correct 72 ms 30520 KB Output is correct
52 Correct 72 ms 30412 KB Output is correct
53 Correct 61 ms 17336 KB Output is correct
54 Correct 63 ms 17380 KB Output is correct
55 Correct 114 ms 23280 KB Output is correct
56 Correct 139 ms 23628 KB Output is correct
57 Correct 177 ms 29504 KB Output is correct
58 Correct 161 ms 29940 KB Output is correct
59 Correct 150 ms 27884 KB Output is correct
60 Correct 149 ms 22916 KB Output is correct
61 Correct 115 ms 26840 KB Output is correct
62 Correct 110 ms 26964 KB Output is correct
63 Correct 115 ms 27008 KB Output is correct
64 Correct 112 ms 25508 KB Output is correct
65 Correct 219 ms 34212 KB Output is correct
66 Correct 165 ms 30476 KB Output is correct
67 Correct 130 ms 26280 KB Output is correct
68 Correct 75 ms 15748 KB Output is correct
69 Correct 63 ms 15836 KB Output is correct
70 Correct 112 ms 26672 KB Output is correct
71 Correct 100 ms 25164 KB Output is correct
72 Correct 4 ms 7372 KB Output is correct
73 Correct 10 ms 8020 KB Output is correct
74 Correct 8 ms 7892 KB Output is correct
75 Correct 4 ms 7376 KB Output is correct
76 Correct 4 ms 7252 KB Output is correct
77 Correct 8 ms 8532 KB Output is correct
78 Correct 10 ms 8020 KB Output is correct
79 Correct 8 ms 8020 KB Output is correct
80 Correct 8 ms 8020 KB Output is correct
81 Correct 5 ms 7368 KB Output is correct
82 Correct 4 ms 7252 KB Output is correct
83 Correct 4 ms 7376 KB Output is correct
84 Correct 4 ms 7372 KB Output is correct
85 Correct 4 ms 7252 KB Output is correct
86 Correct 42 ms 13648 KB Output is correct
87 Correct 38 ms 13728 KB Output is correct
88 Correct 39 ms 13772 KB Output is correct
89 Correct 8 ms 8668 KB Output is correct
90 Correct 8 ms 8020 KB Output is correct
91 Correct 49 ms 13052 KB Output is correct
92 Correct 55 ms 13008 KB Output is correct
93 Correct 4 ms 7380 KB Output is correct
94 Correct 107 ms 31712 KB Output is correct
95 Correct 107 ms 31308 KB Output is correct
96 Correct 121 ms 30880 KB Output is correct
97 Correct 92 ms 30544 KB Output is correct
98 Correct 101 ms 33160 KB Output is correct
99 Correct 155 ms 32016 KB Output is correct
100 Correct 172 ms 29304 KB Output is correct
101 Correct 142 ms 29684 KB Output is correct
102 Correct 122 ms 23024 KB Output is correct
103 Correct 110 ms 26572 KB Output is correct
104 Correct 118 ms 26968 KB Output is correct
105 Correct 134 ms 26444 KB Output is correct
106 Correct 116 ms 25568 KB Output is correct
107 Correct 91 ms 25452 KB Output is correct
108 Correct 75 ms 30512 KB Output is correct
109 Correct 85 ms 30448 KB Output is correct
110 Correct 61 ms 17360 KB Output is correct
111 Correct 61 ms 17484 KB Output is correct
112 Correct 130 ms 23104 KB Output is correct
113 Correct 130 ms 23200 KB Output is correct