#include <bits/stdc++.h>
using namespace std;
int n, k;
vector<int> intervalL;
vector<int> intervalR;
vector<int> depth;
vector<vector<int>> adjList;
vector<vector<int>> check;
int track = 0;
void dfs(int node, int parent){
if (depth[node] == k){
intervalL[node] = track;
intervalR[node] = track;
track++;
}
else{
for (int child : adjList[node]){
if (child == parent) continue;
depth[child] = depth[node]+1;
dfs(child, node);
intervalL[node] = min(intervalL[child], intervalL[node]);
intervalR[node] = max(intervalR[child], intervalR[node]);
}
}
if (node != 0 and intervalR[node] != -1){
check[intervalR[node]].push_back(node);
}
}
int main(){
cin >> n >> k;
adjList.resize(n);
depth.resize(n);
intervalL.resize(n);
intervalR.resize(n);
check.resize(n);
depth[0] = 0;
fill(intervalL.begin(), intervalL.end(), 1000000007);
fill(intervalR.begin(), intervalR.end(), -1);
for (int i = 0; i < n-1; i++){
int a, b;
cin >> a >> b;
adjList[a-1].push_back(b-1);
adjList[b-1].push_back(a-1);
}
if (k*k > n){
cout<<"DA";
return 0;
}
dfs(0, 0);
bool valid = false;
//dp[i][j] - can we use only nodes of depths in subset i such
//that we cannot get to the first j(0 index) nodes at depth K
bool dp[(1<<k)][track];
memset(dp, false, sizeof(dp));
dp[0][0] = true;
for (int i = 0; i < (1<<k); i++){
//can we fill interval [1, j]
for (int j = 0; j < track; j++){
//go through all points going to j
for (int node : check[j]){
//make sure depth is in subset
if ((i&(1<<depth[node]-1)) != 0){
dp[i][j] = dp[i][j] or (dp[i^(1<<depth[node]-1)][intervalL[node]-1]);
}
}
}
if (dp[i][track-1])
valid = true;
}
if (valid)
cout<<"DA";
else
cout<<"NE";
}
Compilation message
burza.cpp: In function 'int main()':
burza.cpp:69:39: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
69 | if ((i&(1<<depth[node]-1)) != 0){
burza.cpp:70:65: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
70 | dp[i][j] = dp[i][j] or (dp[i^(1<<depth[node]-1)][intervalL[node]-1]);
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
18 ms |
1740 KB |
Output is correct |
2 |
Correct |
156 ms |
12236 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
204 KB |
Output is correct |
5 |
Incorrect |
1 ms |
336 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
143 ms |
10700 KB |
Output is correct |
2 |
Correct |
146 ms |
10316 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
332 KB |
Output is correct |
5 |
Incorrect |
146 ms |
10444 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
163 ms |
10572 KB |
Output is correct |
2 |
Correct |
152 ms |
10580 KB |
Output is correct |
3 |
Correct |
1 ms |
332 KB |
Output is correct |
4 |
Correct |
1 ms |
204 KB |
Output is correct |
5 |
Correct |
1 ms |
332 KB |
Output is correct |
6 |
Incorrect |
1 ms |
204 KB |
Output isn't correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
40 ms |
4300 KB |
Output is correct |
2 |
Incorrect |
158 ms |
12616 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
145 ms |
9036 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
141 ms |
11212 KB |
Output is correct |
2 |
Correct |
137 ms |
9800 KB |
Output is correct |
3 |
Correct |
1 ms |
332 KB |
Output is correct |
4 |
Correct |
1 ms |
332 KB |
Output is correct |
5 |
Incorrect |
1 ms |
332 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
154 ms |
10828 KB |
Output is correct |
2 |
Correct |
148 ms |
10188 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
208 KB |
Output is correct |
5 |
Incorrect |
151 ms |
9548 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
37 ms |
3220 KB |
Output is correct |
2 |
Correct |
154 ms |
11596 KB |
Output is correct |
3 |
Correct |
1 ms |
332 KB |
Output is correct |
4 |
Correct |
1 ms |
336 KB |
Output is correct |
5 |
Incorrect |
1 ms |
204 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
17 ms |
1356 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
65 ms |
4228 KB |
Output is correct |
2 |
Correct |
161 ms |
11852 KB |
Output is correct |
3 |
Correct |
1 ms |
204 KB |
Output is correct |
4 |
Correct |
1 ms |
332 KB |
Output is correct |
5 |
Incorrect |
64 ms |
4428 KB |
Output isn't correct |
6 |
Halted |
0 ms |
0 KB |
- |