Submission #234749

#	Time	Username	Problem	Language	Result	Execution time	Memory
234749		muhammad_hokimiyon	Crocodile's Underground City (IOI11_crocodile)	C++14	46 / 100	8 ms	3072 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>

#define fi first
#define se second
#define ll long long
#define dl double long

using namespace std;

const int NN = 2e6 + 7;
const int N = 1e5 + 7;
const int M = 22;
const int mod = 998244353;
const ll inf = 1e18 + 7;
const dl rf = 1e-14;
const int B = sqrt(N);
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

int b[N];
bool used[N];
vector < pair < int , ll > > v[N];

pair < ll , int > dfs( int x )
{
    if( b[x] ){
        return {0 , 0};
    }
    used[x] = true;
    vector < pair < ll , int > > g;
    for( int i = 0; i < (int)v[x].size(); i++ ){
        auto y = v[x][i];
        if( used[y.fi] )continue;
        auto xx = dfs( y.fi );
        xx.fi += y.se;
        //cout << x << " " << y.fi << " " << xx << "\n";
        g.push_back(xx);
    }
    if( (int)g.size() < 2 )return {g[0].fi , 1};
    sort( g.begin() , g.end() );
    if( g[0].se == 0 )return g[1];
    int l = 1;
    while( l < (int)g.size() - 1 && g[l].se == 1 )l += 1;
    return g[l];
}

int travel_plan(int n, int m, int a[][2], int l[], int k, int p[])
{
    for( int i = 0; i < m; i++ ){
        a[i][0] += 1 , a[i][1] += 1;
    }
    for( int i = 0; i < m; i++ ){
        v[a[i][0]].push_back({ a[i][1] , l[i] });
        v[a[i][1]].push_back({ a[i][0] , l[i] });
    }
    for( int i = 0; i < k; i++ ){
        p[i] += 1;
        b[p[i]] = 1;
    }
    return dfs(1).fi;
}

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