제출 #856811

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
856811		Trisanu_Das	Valley (BOI19_valley)	C++17	59 / 100	183 ms	27684 KiB

valley

#pragma GCC optimize("Ofast")
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define int long long
#define ff first
#define ss second
#define pi pair <int, int>
 
int n, s, q, e;
vector<int> d;
int f[100001], l[100001];
pi road[100001];
int dt[1001];
bitset<100001> sh;
vector <pi> g[100001];
 
void dfs(int p, int cur){
    d.pb(cur);
    for(pi i: g[cur]) if(i.ff!=p) dfs(cur, i.ff);
    d.pb(cur);
}
 
void bfs(int rip1, int rip2, int p, int cur){
    for(pi i: g[cur]){
        if(i.ff!=p && (min(cur, i.ff)!=rip1 || max(cur, i.ff)!=rip2)){
            dt[i.ff]=dt[cur]+i.ss;
            bfs(rip1, rip2, cur, i.ff);
        }
    }
}
 
int32_t main(){
    ios_base :: sync_with_stdio(0); cin.tie(); cout.tie();
    cin>>n>>s>>q>>e;
    for(int i=1; i<=n-1; i++){
        int x, y, z;
        cin>>x>>y>>z;
        road[i]={x, y};
        g[x].pb({y, z});
        g[y].pb({x, z});
    }
    for(int i=1; i<=s; i++){
        int x;
        cin>>x;
        sh[x]=1;
    }
    dfs(0, e);
    for(int i=0; i<d.size(); i++){
        l[d[i]]=i+1;
        if(!f[d[i]]) f[d[i]]=i+1;
    }
    if(s==n){
        for(int i=1; i<=q; i++){
            int rd, r;
            cin>>rd>>r;
            int x=road[rd].ff, y=road[rd].ss;
            if(f[x]<=f[r] && l[x]>=l[r] && f[y]<=f[r] && l[y]>=l[r]) cout<<0<<"\n";
            else cout<<"escaped\n";
        }
        return 0;
    }
    for(int i=1; i<=q; i++){
        int rd, r;
        cin>>rd>>r;
        int x=road[rd].ff, y=road[rd].ss;
        if(x>y) swap(x, y);
        if(f[x]<=f[r] && l[x]>=l[r] && f[y]<=f[r] && l[y]>=l[r]){
            for(int j=1; j<=n; j++) dt[j]=1e18;
            dt[r]=0;
            bfs(x, y, 0, r);
            int ans=1e18;
            for(int i=1; i<=n; i++) if(sh[i]) ans=min(ans, dt[i]);
            if(ans!=1e18) cout<<ans<<"\n";
            else cout<<"oo\n";
        }
        else cout<<"escaped\n";
    }
}

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

valley.cpp: In function 'int32_t main()':
valley.cpp:49:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   49 |     for(int i=0; i<d.size(); i++){
      |                  ~^~~~~~~~~

• Subtask 1: 9 / 9
• Subtask 2: 27 / 27
• Subtask 3: 23 / 23
• Subtask 4: 0 / 41