oj.uz

Submission #525105

# Submission time Handle Problem Language Result Execution time Memory
525105 2022-02-10T17:24:16 Z benedict0724 Firefighting (NOI20_firefighting) C++17
0 / 100
 279 ms 25632 KB 
#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef pair<int, ll> pi;
const ll INF = 1e18;
int n; ll k;

vector<pi> adj[300002];
bool chk[300002];
int ans = 0;

ll dfs(int x, int p, ll d)
{
    vector<ll> v;
    ll tmp1 = INF, tmp2 = 0;
    ll D = INF; int t = 1;
    for(auto u : adj[x])
    {
        int i = u.first;
        if(i == p) continue;
        ll c = u.second;
        ll P = dfs(i, x, c);
        if(P > 0 && P < tmp1)
        {
            tmp1 = P;
            t = 1;
        }
        if(P < 0 && -P > tmp2)
        {
            tmp2 = -P;
            t = -1;
        }
    }

    if(tmp1 == INF && tmp2 == 0) { D = INF; }
    else if(tmp1 == INF) { D = tmp2; t = -1; }
    else if(tmp1 > tmp2) { D = tmp1; t = 1; }
    else { D = tmp2; t = -1; }

    if(D == INF)
    {
        if(d > k)
        {
            chk[x] = 1; ans++;
            return 2*k+2;
        }
        return -(d+k+1);
    }

    if(D > k && D + d > 2*k+1)
    {
        chk[x] = 1; ans++;
        return min(d, k+1);
    }
    else if(D > k) return t*(D+d);
    if(D + d > k) return t*(k+1);
    else return t*(D+d);
}

int main()
{
    ios::sync_with_stdio(false); cin.tie(NULL);
    cin >> n >> k;
    for(int i=1;i<n;i++)
    {
        int a, b; ll D; cin >> a >> b >> D;
        adj[a].push_back({b, D});
        adj[b].push_back({a, D});
    }

    ll P = dfs(1, 0, 0);
    if(P < 0) P = -P;
    else if(P == 0 && chk[1] == 0) { chk[1] = 1; ans++; }
    else if(P > k && chk[1] == 0) { chk[1] = 1; ans++; }

    assert(ans <= n);
    cout << ans << "\n";

    for(int i=1;i<=n;i++) if(chk[i]) cout << i << " ";
}

Subtask #1
0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 265 ms 25632 KB Output is correct
2 Correct 270 ms 25540 KB Output is correct
3 Correct 72 ms 14020 KB Output is correct
4 Correct 265 ms 25028 KB Output is correct
5 Correct 4 ms 7244 KB Output is correct
6 Incorrect 4 ms 7244 KB Integer parameter [name=F] equals to 0, violates the range [1, 1]
7 Halted 0 ms 0 KB -

Subtask #2
0 / 5.0

# Verdict Execution time Memory Grader output
1 Incorrect 3 ms 7244 KB Integer parameter [name=F] equals to 0, violates the range [1, 17]
2 Halted 0 ms 0 KB -

Subtask #3
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 7244 KB Output is correct
2 Correct 4 ms 7244 KB Output is correct
3 Incorrect 3 ms 7244 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #4
0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 279 ms 25620 KB Output is correct
2 Incorrect 119 ms 17116 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #5
0 / 26.0

# Verdict Execution time Memory Grader output
1 Incorrect 5 ms 7500 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 38.0

# Verdict Execution time Memory Grader output
1 Incorrect 242 ms 22980 KB Output isn't correct
2 Halted 0 ms 0 KB -