oj.uz

Submission #118740

# Submission time Handle Problem Language Result Execution time Memory
118740 2019-06-19T14:33:10 Z someone_aa Highway Tolls (IOI18_highway) C++17
11 / 100
 1500 ms 36536 KB 
#include "highway.h"
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define mp make_pair
using namespace std;
const int maxn = 100100;
int n, m;
vector<int>w;
vector<int>gs[maxn];
vector<pair<int,int> > g[maxn];

vector<int>gt[maxn];

int dist[maxn][2], parent[maxn][2];
bool visited[maxn];

ll dista;

map<pair<int,int>, int> ind;

void bfs(int st, int d) {
    memset(visited,false,sizeof(visited));
    queue<int>q;
    q.push(st);

    parent[st][d] = -1;
    visited[st] = true;

    while(!q.empty()) {
        int curr = q.front();
        q.pop();

        for(auto i:g[curr]) {
            if(!visited[i.first]) {
                visited[i.first] = true;
                dist[i.first][d] = dist[curr][d] + 1;
                parent[i.first][d] = i.second;
                q.push(i.first);
            }
        }
    }
}

vector<pair<int,int> > sta, stb;

void preprocess() {
    for(int i=0;i<n;i++) {
        if(dist[i][0] == dist[i][1]) continue; // skip this node

        if(dist[i][0] < dist[i][1]) {
            for(int j:gs[i]) {
                if(dist[j][0] < dist[j][1]) {
                    sta.pb(mp(min(i, j), max(i, j)));
                }
            }
        }
        else {
            for(int j:gs[i]) {
                if(dist[j][0] > dist[j][1]) {
                    stb.pb(mp(min(i, j), max(i, j)));
                }
            }
        }
    }

    sort( sta.begin(), sta.end() );
    sta.erase( unique( sta.begin(), sta.end() ), sta.end() );


    sort( stb.begin(), stb.end() );
    stb.erase( unique( stb.begin(), stb.end() ), stb.end() );
}


int solve(int root, int d) {
    // build bfs tree
    //memset(visited,false,sizeof(visited));
    //memset(parent,-1,sizeof(parent));

    vector<pair<int,int> > edges;
    if(d == 0) edges = sta;
    else edges = stb;

    for(auto i:edges) {
        gt[i.first].pb(i.second);
        gt[i.second].pb(i.first);
    }

    vector<int>v, v2;

    queue<int>q;
    q.push(root);

    visited[root] = true;
    parent[root][d] = -1;

    while(!q.empty()) {
        int curr = q.front();
        q.pop();

        v.pb(parent[curr][d]);
        v2.pb(curr);
        for(auto i:gt[curr]) {
            if(!visited[i]) {
                visited[i] = true;
                parent[i][d] = curr;
                dist[i][d] = dist[curr][d] + 1;
                q.push(i);
            }
        }
    }

    if(v.size() == 1) return v2[0];
    // perform binary search

    for(int i=0;i<v.size();i++) {
        if(v[i] == -1) continue; // root
        int cind = ind[mp(v[i], v2[i])];
        w[cind] = 1;
    }

    int li = 0, ri = v.size() - 1;

    while(li < ri) {
        int mid = (li + ri) / 2;

        // all before mid put them as low cost and other as high cost
        for(int i=0;i<=mid;i++) {
            if(v[i] == -1) continue; // root
            int cind = ind[mp(v[i], v2[i])];
            w[cind] = 0;
        }

        ll cdist = ask(w);
        for(int i=0;i<=mid;i++) {
            if(v[i] == -1) continue; // root
            int cind = ind[mp(v[i], v2[i])];
            w[cind] = 1;
        }

        if(cdist == dista) {
            ri = mid;
        }
        else {
            li = mid + 1;
        }
    }

    for(int i=0;i<v.size();i++) {
        if(v[i] == -1) continue; // root
        int cind = ind[mp(v[i], v2[i])];
        w[cind] = 0;
    }

    return v2[li];
}

void find_pair(int N, std::vector<int> U, std::vector<int> V, int A, int B) {
    n = N; m = U.size();
    int li=0, ri=m-1;

    for(int i=0;i<m;i++) {
        ind[mp(U[i], V[i])] = ind[mp(V[i], U[i])] = i;
        gs[U[i]].pb(V[i]);
        gs[V[i]].pb(U[i]);

        g[U[i]].pb(mp(V[i], i));
        g[V[i]].pb(mp(U[i], i));

        w.pb(0);
    }

    dista = ask(w);
    while(li<ri) {
        int mid = (li+ri)/2;
        for(int i=li;i<=mid;i++) {
            w[i] = 1;
        }
        ll distb = ask(w);
        for(int i=li;i<=mid;i++) {
            w[i] = 0;
        }

        if(dista == distb) {
            li = mid + 1;
        }
        else {
            ri = mid;
        }
    }
    int x = li;
    int u = U[x], v = V[x];

    //cout<<u<<" "<<v<<"\n";

    bfs(u, 0);
    bfs(v, 1);
    preprocess();

    memset(visited,false,sizeof(visited));
    //cout<<"Ans: "<<xx<<" "<<yy<<"\n";
    answer(solve(u, 0), solve(v, 1));
}

Compilation message

highway.cpp: In function 'int solve(int, int)':
highway.cpp:117:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<v.size();i++) {
                 ~^~~~~~~~~
highway.cpp:150:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<v.size();i++) {
                 ~^~~~~~~~~

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 7400 KB Output is correct
2 Correct 9 ms 7544 KB Output is correct
3 Correct 9 ms 7392 KB Output is correct
4 Correct 9 ms 7416 KB Output is correct
5 Correct 8 ms 7404 KB Output is correct
6 Correct 8 ms 7396 KB Output is correct
7 Correct 8 ms 7480 KB Output is correct
8 Correct 9 ms 7400 KB Output is correct
9 Correct 9 ms 7416 KB Output is correct
10 Correct 8 ms 7428 KB Output is correct
11 Correct 9 ms 7544 KB Output is correct
12 Correct 8 ms 7396 KB Output is correct

Subtask #2
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 7676 KB Output is correct
2 Correct 60 ms 10444 KB Output is correct
3 Correct 1460 ms 34700 KB Output is correct
4 Execution timed out 1651 ms 34772 KB Time limit exceeded
5 Halted 0 ms 0 KB -

Subtask #3
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 36 ms 10276 KB Output is correct
2 Correct 84 ms 13084 KB Output is correct
3 Correct 119 ms 16204 KB Output is correct
4 Correct 418 ms 32636 KB Output is correct
5 Correct 368 ms 32732 KB Output is correct
6 Correct 723 ms 33200 KB Output is correct
7 Correct 363 ms 33980 KB Output is correct
8 Correct 574 ms 32520 KB Output is correct
9 Correct 356 ms 33560 KB Output is correct

Subtask #4
0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 12 ms 7804 KB Output is correct
2 Correct 62 ms 10456 KB Output is correct
3 Correct 498 ms 28828 KB Output is correct
4 Correct 791 ms 34804 KB Output is correct
5 Correct 672 ms 34896 KB Output is correct
6 Correct 686 ms 34524 KB Output is correct
7 Correct 630 ms 34592 KB Output is correct
8 Correct 635 ms 34484 KB Output is correct
9 Correct 1183 ms 34656 KB Output is correct
10 Correct 1000 ms 34736 KB Output is correct
11 Execution timed out 1610 ms 33992 KB Time limit exceeded
12 Halted 0 ms 0 KB -

Subtask #5
0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 79 ms 10524 KB Output is correct
2 Correct 57 ms 10896 KB Output is correct
3 Execution timed out 1909 ms 36352 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #6
0 / 31.0

# Verdict Execution time Memory Grader output
1 Correct 66 ms 10468 KB Output is correct
2 Correct 107 ms 10880 KB Output is correct
3 Correct 681 ms 34404 KB Output is correct
4 Incorrect 1115 ms 36536 KB Output is incorrect: {s, t} is wrong.
5 Halted 0 ms 0 KB -