Submission #370841

#	Time	Username	Problem	Language	Result	Execution time	Memory
370841		azberjibiou	Designated Cities (JOI19_designated_cities)	C++17	0 / 100	351 ms	524292 KiB

designated_cities

#include <bits/stdc++.h>
#define fir first
#define sec second
#define ll long long
#define pll pair<ll, ll>
using namespace std;
const int mxN=200020;
const ll INF=1000000000000000001;
typedef struct line{
    int s, e;
    ll val;
}line;
int N, Q;
map <int, pair<int, ll>> v[mxN];
vector <pll> rv[mxN];
vector <line> E;
ll ans[mxN];
ll tot;
ll sub[mxN], rsub[mxN];
void dfs(int now, int pre)
{
    for(auto iter=v[now].begin();iter!=v[now].end();iter++)
    {
        if(iter->fir==pre)  continue;
        dfs(iter->sec.fir, now);
        sub[now]+=sub[iter->fir];
        rsub[now]+=rsub[iter->fir];
        sub[now]+=iter->sec.sec;
    }
    for(pll ele : rv[now])
    {
        if(ele.fir!=pre)    rsub[now]+=ele.sec;
        if(ele.fir!=pre)    tot+=ele.sec;
    }
}
void solv1()
{
    ans[1]=INF;
    dfs(1, -1);
    for(int i=1;i<=N;i++)
    {
        ans[1]=min(ans[1], tot-rsub[i]+sub[i]);
    }
}
int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
    cin >> N;
    for(int i=0;i<N-1;i++)
    {
        int a, b, c, d;
        cin >> a >> b >> c >> d;
        v[a].insert({b, {2*i, d}});
        v[b].insert({a, {2*i+1, c}});
        rv[a].push_back({b, c});
        rv[b].push_back({a, d});
    }
    solv1();
    cin >> Q;
    while(Q--)
    {
        int a;
        cin >> a;
        cout << ans[a] << '\n';
    }
}

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