Submission #955732

# Submission time Handle Problem Language Result Execution time Memory
955732 2024-03-31T11:25:59 Z TimAni Burza (COCI16_burza) C++17
0 / 160
237 ms 1400 KB
//Bitmask DP Solution
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define ms(x, a) memset(x, a, sizeof(x))
const int INF = 0x3f3f3f3f;
//===========================================
const int MAX = 405;
vector<int> adj[MAX];
int dep[MAX], dp[1<<19], l[MAX], r[MAX], curr, n, k;

void dfs(int v, int p){
    if (dep[v] == k){ l[v] = r[v] = ++curr; return; }
    for (int u: adj[v]){
        if (u == p) continue;
        dep[u] = dep[v]+1;
        dfs(u, v);
        l[v] = min(l[v], l[u]);
        r[v] = max(r[v], r[u]);
    }
}

int main(){
    cin.tie(0)->sync_with_stdio(0);
    cin >> n >> k;
    ms(l, INF); ms(r, 0);
    if (k >= 20){ cout << "NE\n"; return 0; }
    for (int i = 1; i < n; i++){
        int a, b; cin >> a >> b;
        adj[a].pb(b);
        adj[b].pb(a);
    }
    dfs(1, -1);
    for (int mask = 1; mask < 1<<k; mask++){
        dp[mask] = -INF;
        for (int i = 1; i <= n; i++){
            if (!dep[i] || !(mask&(1<<(dep[i]-1)))) continue;
            if (dp[mask^(1<<(dep[i]-1))]+1 >= l[i]) dp[mask] = max({dp[mask], r[i], dp[mask^(1<<(dep[i]-1))]});
        }
    }
    cout << (dp[(1<<k)-1] == curr? "DA\n" : "NE\n");
}
# Verdict Execution time Memory Grader output
1 Correct 23 ms 348 KB Output is correct
2 Correct 237 ms 788 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 225 ms 1076 KB Output is correct
2 Correct 222 ms 1340 KB Output is correct
3 Incorrect 0 ms 344 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 237 ms 908 KB Output is correct
2 Correct 227 ms 776 KB Output is correct
3 Incorrect 1 ms 600 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 51 ms 540 KB Output is correct
2 Correct 229 ms 868 KB Output is correct
3 Incorrect 0 ms 344 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 230 ms 1400 KB Output is correct
2 Correct 228 ms 852 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 228 ms 888 KB Output is correct
2 Correct 221 ms 812 KB Output is correct
3 Incorrect 0 ms 344 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 221 ms 948 KB Output is correct
2 Correct 223 ms 1080 KB Output is correct
3 Incorrect 0 ms 344 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 49 ms 596 KB Output is correct
2 Correct 233 ms 904 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 23 ms 348 KB Output is correct
2 Correct 235 ms 808 KB Output is correct
3 Incorrect 0 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 105 ms 516 KB Output is correct
2 Correct 221 ms 808 KB Output is correct
3 Incorrect 1 ms 348 KB Output isn't correct
4 Halted 0 ms 0 KB -