oj.uz

Submission #655013

# Submission time Handle Problem Language Result Execution time Memory
655013 2022-11-02T12:50:44 Z DJeniUp Designated Cities (JOI19_designated_cities) C++17
0 / 100
 589 ms 66636 KB 
#include "bits/stdc++.h"
//#pragma GCC optimize("O3")

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>pairll;
typedef long double ld;

#define fr first
#define sc second
#define pb push_back
#define INF 100000000000000007
#define I 100000000000007
#define N 100007
#define endl '\n'
#define MOD 998244353

ll n,res,resx,d[2*N],sz;

vector<pairll>v[2*N];

map<ll,ll>m[2*N];

priority_queue<ll>q,q1;

ll S(ll x,ll y){
    ll rs=0;
    for(int i=0;i<v[x].size();i++){
        if(v[x][i].fr!=y){
            rs+=v[x][i].sc+S(v[x][i].fr,x);
        }
    }
    return rs;
}

void D(ll x,ll y,ll z){
    if(res>z){
        res=z;
        resx=x;
        sz=1;
    }else if(res==z)sz++;
    for(int i=0;i<v[x].size();i++){
        if(v[x][i].fr!=y){
            D(v[x][i].fr,x,z-v[x][i].sc+m[v[x][i].fr][x]);
        }
    }
    return ;
}

ll R(ll x,ll y){
    ll k=0;
    for(int i=0;i<v[x].size();i++){
        if(v[x][i].fr!=y){
            ll r=R(v[x][i].fr,x)+v[x][i].sc;
            if(k<r)swap(k,r);
            if(r!=0 && x!=resx)q.push(r);
            else if(r!=0)q1.push(r);
        }
    }
    return k;
}

int main()
{
    cin>>n;
    for(int i=1;i<n;i++){
        ll x,y,z,z1;
        cin>>x>>y>>z>>z1;
        v[x].pb({y,z});
        v[y].pb({x,z1});
        m[x][y]=z;
        m[y][x]=z1;
    }
    res=INF;
    D(1,1,S(1,1));
    if(sz>1)exit(1);
    ll k=R(resx,0);
    q1.push(k);
    ll x=res;
    //cout<<"! "<<res<<" "<<resx<<endl;
    q1.push(0);
    res-=q1.top();
    q1.pop();
    res-=q1.top();
    q1.pop();
    d[2]=res;
    while(q1.size()>0){
        q.push(q1.top());
        q1.pop();
    }
    for(int i=3;i<=n;i++){
        if(q.size()>0){
            res-=q.top();
            //cout<<i<<" "<<q.top()<<endl;
            q.pop();
        }
        d[i]=res;
    }
    d[1]=x;
    ll Q;
    cin>>Q;
    for(int i=1;i<=Q;i++){
        cin>>x;
        cout<<d[x]<<endl;
    }

    return 0;
}

Compilation message

designated_cities.cpp: In function 'll S(ll, ll)':
designated_cities.cpp:30:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |     for(int i=0;i<v[x].size();i++){
      |                 ~^~~~~~~~~~~~
designated_cities.cpp: In function 'void D(ll, ll, ll)':
designated_cities.cpp:44:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   44 |     for(int i=0;i<v[x].size();i++){
      |                 ~^~~~~~~~~~~~
designated_cities.cpp: In function 'll R(ll, ll)':
designated_cities.cpp:54:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   54 |     for(int i=0;i<v[x].size();i++){
      |                 ~^~~~~~~~~~~~

Subtask #1
0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14292 KB Output is correct
2 Correct 7 ms 14292 KB Output is correct
3 Correct 8 ms 14292 KB Output is correct
4 Runtime error 7 ms 14320 KB Execution failed because the return code was nonzero
5 Halted 0 ms 0 KB -

Subtask #2
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14320 KB Output is correct
2 Correct 562 ms 51948 KB Output is correct
3 Correct 589 ms 66636 KB Output is correct
4 Runtime error 482 ms 49612 KB Execution failed because the return code was nonzero
5 Halted 0 ms 0 KB -

Subtask #3
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14292 KB Output is correct
2 Incorrect 550 ms 52028 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14292 KB Output is correct
2 Correct 7 ms 14292 KB Output is correct
3 Correct 8 ms 14292 KB Output is correct
4 Runtime error 7 ms 14320 KB Execution failed because the return code was nonzero
5 Halted 0 ms 0 KB -

Subtask #5
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14320 KB Output is correct
2 Correct 562 ms 51948 KB Output is correct
3 Correct 589 ms 66636 KB Output is correct
4 Runtime error 482 ms 49612 KB Execution failed because the return code was nonzero
5 Halted 0 ms 0 KB -

Subtask #6
0 / 44.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 14292 KB Output is correct
2 Correct 7 ms 14292 KB Output is correct
3 Correct 8 ms 14292 KB Output is correct
4 Runtime error 7 ms 14320 KB Execution failed because the return code was nonzero
5 Halted 0 ms 0 KB -