oj.uz

Submission #383136

# Submission time Handle Problem Language Result Execution time Memory
383136 2021-03-28T21:53:45 Z Mastermind Traffic (IOI10_traffic) C++17
0 / 100
 2 ms 492 KB 
#include <bits/stdc++.h>

#include "traffic.h"
using namespace std;

#define pb push_back
#define f first
#define s second

void rawSummer(int root, int N, vector<bool> vis, vector<int> adj[],
               vector<long long> rawSum, int parent = -1) {
    vis[root] = true;
    for (int next : adj[root]) {
        if (!vis[root]) {
            rawSummer(next, N, vis, adj, rawSum, root);
        }
    }

    for (int next : adj[root]) {
        if (next != parent) {
            rawSum[root] += rawSum[next];
        }
    }
}

void weightSummer(int root, int N, vector<bool> vis, vector<int> adj[],
                  vector<long long> weightSum, vector<long long> rawSum,
                  int parent = -1) {
    vis[root] = true;
    for (int next : adj[root]) {
        if (!vis[root]) {
            weightSummer(next, N, vis, adj, weightSum, rawSum, root);
        }
    }

    for (int next : adj[root]) {
        if (next != parent) {
            weightSum[root] += weightSum[next] + rawSum[next];
        }
    }
}

int LocateCentre(int N, int pp[], int S[], int D[]) {
    vector<int> adj[N];

    for (int i = 0; i < N; i++) {
        adj[S[i]].pb(D[i]);
        adj[D[i]].pb(S[i]);
    }

    int start = 0;
    for (int i = 1; i < N; i++) {
        if (adj[i].size() > adj[start].size()) start = i;
    }

    // # Construct the Graphs
    vector<long long> rawSum(N);
    vector<long long> weightSum(N);
    vector<bool> vis(N);

    for (int i = 0; i < N; i++) {
        rawSum[i] = pp[i];
    }

    rawSummer(start, N, vis, adj, rawSum);
    vis = *new vector<bool>(N);
    weightSummer(start, N, vis, adj, weightSum, rawSum);

    // # Now Check Each One, Starting From the Root
    // _ Use BFS
    vis = *new vector<bool>(N);
    queue<pair<int, long long>> q;
    q.push({start, weightSum[start] + rawSum[start]});

    long long cmin = LLONG_MAX;
    int index = -1;

    while (!q.empty()) {
        pair<int, long long> cur = q.front();
        q.pop();
        vis[cur.f] = true;
        long long congestion = cur.s;

        congestion += (rawSum[start] - rawSum[cur.f]);
        congestion -= (rawSum[cur.f]);

        if (congestion < cmin) {
            cmin = congestion;
            index = cur.f;
        }

        for (int next : adj[cur.f]) {
            if (!vis[next]) {
                q.push({next, congestion});
            }
        }
    }

    cout << index << endl;
}

Compilation message

traffic.cpp: In function 'int LocateCentre(int, int*, int*, int*)':
traffic.cpp:100:1: warning: no return statement in function returning non-void [-Wreturn-type]
  100 | }
      | ^

Subtask #1
0 / 25.0

# Verdict Execution time Memory Grader output
1 Runtime error 2 ms 492 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #2
0 / 25.0

# Verdict Execution time Memory Grader output
1 Runtime error 2 ms 492 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #3
0 / 25.0

# Verdict Execution time Memory Grader output
1 Runtime error 2 ms 492 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #4
0 / 25.0

# Verdict Execution time Memory Grader output
1 Runtime error 2 ms 492 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -