제출 #1102640

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1102640		asli_bg	Crocodile's Underground City (IOI11_crocodile)	C++11	46 / 100	229 ms	262144 KiB

crocodile

#pragma GCC optimize("O3,unroll-loops")
#include "crocodile.h"
 
#include<bits/stdc++.h>
using namespace std;
 
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define FOR(i,a) for(int i=0;i<(a);i++)
#define FORE(i,a,b) for(int i=(a);i<(b);i++)
 
typedef pair<int,int> pii;
typedef vector<int> vi;
typedef vector<bool> vb;
typedef vector<pii> vii;
 
#define cont(x) for(auto el:x) {cout<<el<<' ';} cout<<endl;
#define endl '\n'
#define contp(x) for(auto el:x) {cout<<el.fi<<'-'<<el.se<<' ';} cout<<endl;
#define pb push_back
 
#define sp <<' '<<
 
#define DEBUG(x) {cout<<(#x) sp x<<endl;}
 
const int MAXN=1e5+5;
const int INF=1e9;
 
int n,m,k;
vii adj[MAXN];
int mn[MAXN],mn2[MAXN];
bool cikis[MAXN];

void dfs(int nd,int ata){
    for(auto el:adj[nd]){
        int kom=el.fi;
        int cost=el.se;
        if(kom==ata) continue;
        dfs(kom,nd);
        //mn[nd], mn2[nd] / mn[kom], mn2[kom]
        vi temp;
        temp.pb(mn[nd]);
        temp.pb(mn2[nd]);
        temp.pb(mn2[kom]+cost);
        sort(all(temp));

        mn[nd]=temp[0];
        mn2[nd]=temp[1];
    }
}

int travel_plan(int N, int M, int R[][2], int L[], int K, int P[]){
    n=N;
    m=M;
    k=K;
    FOR(i,m){
        int bir=R[i][0];
        int iki=R[i][1];
        bir++;iki++;
        adj[bir].pb({iki,L[i]});
        adj[iki].pb({bir,L[i]});
    }
 
    FOR(i,k){
        P[i]++;
        cikis[P[i]]=true;
    }

    FORE(i,1,n+1){
        if(cikis[i]) {mn[i]=mn2[i]=0;}
        else mn[i]=mn2[i]=INF;
    }

    dfs(1,0);

    return mn2[1];
}

• Subtask 1: 46 / 46
• Subtask 2: 0 / 43
• Subtask 3: 0 / 11