oj.uz

Submission #522644

# Submission time Handle Problem Language Result Execution time Memory
522644 2022-02-05T10:01:08 Z Andy__Andy__ Sightseeing (NOI14_sightseeing) C++17
25 / 25
 2202 ms 250428 KB 
#include <bits/stdc++.h>
#define ll long long
#define int ll
using namespace std;

ifstream f ("test.in");
ofstream g ("test.out");
int teste=1;
int n,m,q;
int dp[500005];
int tata[500005];
int rang[500005];

vector < array < int ,2 > > G[500005];

int rep(int nod)
{
    if(tata[nod] == nod) return nod;
    return (tata[nod] = rep(tata[nod]));
}

void comb(int i,int j,int cost)
{
    int ri = rep(i);
    int rj = rep(j);

    if(ri != rj)
    {
        if(rang[ri]<rang[rj]) swap(ri,rj);
        tata[rj] = ri;
        rang[ri] += rang[rj];
        G[i].push_back({cost,j});
        G[j].push_back({cost,i});
    }
}
vector < array < int ,3 > > E;


void dfs(int nod,int tata)
{
    for(auto x:G[nod])
    {
        int vecin = x[1];
        int cost = x[0];
        if(vecin==tata) continue;
        dp[vecin] = min(dp[nod],cost);
        dfs(vecin,nod);
    }
}

void solve()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    cin>>n>>m>>q;

    for(int i=1;i<=n;++i)
    {
        tata[i]=i;
        rang[i]=1;
    }

    for(int i=1;i<=m;++i)
    {
        int a,b,c;
        cin>>a>>b>>c;
        E.push_back({-c,a,b});
    }

    sort(E.begin(),E.end());

    for(int i=0;i<E.size();++i)
    {
        comb(E[i][1],E[i][2],-E[i][0]);
    }
    dp[1] = 1e9;
    dfs(1,-1);

    for(;q--;)
    {
        int x;
        cin>>x;
        cout<<dp[x]<< '\n';
    }


}


main()
{
    while(teste--)
    {
        solve();
    }

    return 0;
}

Compilation message

sightseeing.cpp: In function 'void solve()':
sightseeing.cpp:74:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::array<long long int, 3> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   74 |     for(int i=0;i<E.size();++i)
      |                 ~^~~~~~~~~
sightseeing.cpp: At global scope:
sightseeing.cpp:92:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   92 | main()
      | ^~~~

Subtask #1
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 12072 KB Output is correct
2 Correct 7 ms 12072 KB Output is correct

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 12364 KB Output is correct
2 Correct 7 ms 12088 KB Output is correct

Subtask #3
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 33 ms 16768 KB Output is correct
2 Correct 29 ms 16168 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1506 ms 116560 KB Output is correct
2 Correct 2202 ms 250428 KB Output is correct