oj.uz

Submission #562594

# Submission time Handle Problem Language Result Execution time Memory
562594 2022-05-14T17:43:03 Z Leo121 Ideal city (IOI12_city) C++14
55 / 100
 136 ms 5908 KB 
#include <bits/stdc++.h>
#define forn(i, a, b) for(int i = int (a); i <= int(b); ++ i)
#define fora(i, a, b) for(int i = int (a); i >= int(b); -- i)
#define pb push_back
#define fi first
#define se second
#define mp make_pair
using namespace std;
typedef pair<int, int> pii;
typedef long long ll;
typedef vector<int> vi;

const int maxn = 1e5;
int distancia[maxn];
vi graph[maxn];
ll mod = 1e9;
ll res = 0;
int mx[4] = {0, 0, 1, -1};
int my[4] = {1, -1, 0, 0};
map<pii, int> mapa;
queue<int> q;
map<int, int> x;
map<int, int> y;
ll arrex[maxn + 2];
ll arrey[maxn + 2];

void bfs(int ind){
    int u;
    while(!q.empty()){
        u = q.front();
        q.pop();
        for(auto v : graph[u]){
            if(distancia[v] == -1){
                distancia[v] = distancia[u] + 1;
                if(v > ind){
                    res += (ll) distancia[v];
                    res %= mod;
                }
                q.push(v);
            }
        }
    }
}

int DistanceSum(int N, int *X, int *Y) {
    if(N <= 2000){
        forn(i, 0, N - 1){
            mapa[mp(X[i], Y[i])] = i;
        }
        forn(i, 0, N - 1){
            forn(j, 0, 3){
                if(mapa.count(mp(X[i] + mx[j], Y[i] + my[j]))){
                    graph[i].pb(mapa[mp(X[i] + mx[j], Y[i] + my[j])]);
                }
            }
        }
        forn(i, 0, N - 1){
            memset(distancia, -1, sizeof(distancia));
            distancia[i] = 0;
            q.push(i);
            bfs(i);
        }
    }
    else{
        forn(i, 0, N - 1){
            x[X[i]] ++;
            y[Y[i]] ++;
        }
        int longx = 0, longy = 0;
        for(auto i : x){
            arrex[++ longx] = (ll) i.se;
        }
        for(auto i : y){
            arrey[++ longy] = (ll) i.se;
        }
        ll suma = 0;
        forn(i, 2, longx){
            suma += arrex[i] * ((ll) i - 1LL);
        }
        res = suma * arrex[1];
        res %= mod;
        fora(i, longx - 1, 2){
            arrex[i] += arrex[i + 1];
        }
        forn(i, 2, longx){
            suma -= arrex[i];
            res += suma * (arrex[i] - arrex[i + 1]);
            res %= mod;
        }
        suma = 0;
        forn(i, 2, longy){
            suma += arrey[i] * ((ll) i - 1LL);
        }
        res += suma * arrey[1];
        res %= mod;
        fora(i, longy - 1, 2){
            arrey[i] += arrey[i + 1];
        }
        forn(i, 2, longy){
            suma -= arrey[i];
            res += suma * (arrey[i] - arrey[i + 1]);
            res %= mod;
        }
    }
    return (int) res;
}

Subtask #1
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 3028 KB Output is correct
2 Correct 2 ms 3028 KB Output is correct
3 Correct 2 ms 3028 KB Output is correct
4 Correct 3 ms 3028 KB Output is correct
5 Correct 3 ms 3028 KB Output is correct
6 Correct 5 ms 3028 KB Output is correct
7 Correct 5 ms 3028 KB Output is correct
8 Correct 4 ms 3028 KB Output is correct
9 Correct 4 ms 3068 KB Output is correct
10 Correct 5 ms 3028 KB Output is correct
11 Correct 4 ms 3028 KB Output is correct

Subtask #2
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 40 ms 3156 KB Output is correct
2 Correct 46 ms 3156 KB Output is correct
3 Correct 79 ms 3196 KB Output is correct
4 Correct 74 ms 3192 KB Output is correct
5 Correct 125 ms 3156 KB Output is correct
6 Correct 121 ms 3264 KB Output is correct
7 Correct 136 ms 3252 KB Output is correct
8 Correct 128 ms 3156 KB Output is correct
9 Correct 118 ms 3276 KB Output is correct
10 Correct 115 ms 3248 KB Output is correct

Subtask #3
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 2804 KB Output is correct
2 Correct 7 ms 3056 KB Output is correct
3 Correct 15 ms 3524 KB Output is correct
4 Correct 13 ms 3540 KB Output is correct
5 Correct 30 ms 4344 KB Output is correct
6 Correct 27 ms 4300 KB Output is correct
7 Correct 30 ms 4552 KB Output is correct
8 Correct 28 ms 4208 KB Output is correct
9 Correct 26 ms 4392 KB Output is correct
10 Correct 36 ms 5908 KB Output is correct

Subtask #4
0 / 45.0

# Verdict Execution time Memory Grader output
1 Incorrect 7 ms 2880 KB Output isn't correct
2 Halted 0 ms 0 KB -