제출 #472659

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
472659		ITO	Designated Cities (JOI19_designated_cities)	C++11	0 / 100	620 ms	32200 KiB

designated_cities

#include<bits/stdc++.h>
#define mp make_pair
#define no cout << "ickgf\n"
using namespace std;
typedef long long ll;
int n, mi1;
ll an[200001], ma = 0;
vector<pair<int, pair<ll, ll>>> e[200001];
priority_queue<pair<ll, int>> pq[200001];
pair<int, int> eat(int x, int p){
    pair<ll, ll> re = {0ll, 0ll}, tm;
    for (pair<int, pair<ll, ll>> pi : e[x]) {
        if (pi.first != p) {
            tm = eat(pi.first, x);
            re.first += tm.first + pi.second.second;
            re.second = max(re.second, tm.second + pi.second.first - pi.second.second);
        }
    }
    return re;
}
pair<ll, ll> nom(int x, int p, ll ba) {
    pair<ll, ll> m1 = {0, 0}, m2 = {0, 0}, tm;
    for (pair<int, pair<ll, ll>> pi : e[x]) {
        if (pi.first != p) {
            tm = nom(pi.first, x, ba + pi.second.first - pi.second.second);
            tm.first += (ll) pi.second.first;
            if (tm.first > m1.first) {
                m2 = m1, m1 = tm;
            } else if (tm.first > m2.first) {
                m2 = tm;
            }
        }
    }
    if (m2.first > 0ll) {
        ba += m1.first + m2.first;
        if (ba > ma) {
            ma = ba, mi1 = m1.second;
        }
    } else if (m1.first == 0ll) {
        m1.second = x;
    }
    return m1;
}
ll yum(int x, int p){
    if (e[x].size() == 1 && p != 0) {
        return 0;
    }
    for (pair<int, pair<ll, ll>> pi : e[x]) {
        if (pi.first != p) {
            pq[x].push({yum(pi.first, x) + pi.second.first, pi.first});
        }
    }
    return pq[x].top().first;
}
void mhm(int x) {
    if (!pq[x].empty()) {
        pair<ll, int> ppp = pq[x].top();
        pq[x].pop();
        mhm(ppp.second);
    }
    while (!pq[x].empty()) {
        pq[mi1].push(pq[x].top());
        pq[x].pop();
    }
}
int main(){
    int a, b, c, d, ej, q;
    ll tot = 0;
    cin >> n;
    for (int i = 1; i < n; i++) {
        cin >> a >> b >> c >> d;
        e[a].push_back({b, {c, d}});
        e[b].push_back({a, {d, c}});
        tot += (ll) c + d;
    }
    if (n == 2) {
        an[1] = min(c, d);
    } else {
        int sta = 1;
        if (e[1].size() == 1) {
            sta = e[1][0].first;
        }
        pair<ll, ll> re = eat(sta, 0), re2 = nom(sta, 0, re.first);
        an[1] = tot - re.first - re.second;
        yum(mi1, 0);
        tot -= ma;
        an[2] = tot;
        pair<ll, int> ppp;
        ppp = pq[mi1].top();
        pq[mi1].pop();
        mhm(ppp.second);
        for (int i = 3; tot > 0 && i <= n; i++) {
            ppp = pq[mi1].top();
            pq[mi1].pop();
            mhm(ppp.second);
            tot -= ppp.first;
            an[i] = tot;
        }
    }
    cin >> q;
    for (int i = 0; i < q; i++) {
        cin >> ej;
        cout << an[ej] << '\n';
    }
    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

designated_cities.cpp: In function 'int main()':
designated_cities.cpp:83:40: warning: variable 're2' set but not used [-Wunused-but-set-variable]
   83 |         pair<ll, ll> re = eat(sta, 0), re2 = nom(sta, 0, re.first);
      |                                        ^~~

• Subtask 1: 0 / 6
• Subtask 2: 0 / 7
• Subtask 3: 0 / 9
• Subtask 4: 0 / 17
• Subtask 5: 0 / 17
• Subtask 6: 0 / 44