oj.uz

Submission #575534

# Submission time Handle Problem Language Result Execution time Memory
575534 2022-06-10T21:05:17 Z FatihSolak Ideal city (IOI12_city) C++17
0 / 100
 1000 ms 7312 KB 
#include <bits/stdc++.h>
#define N 100005
using namespace std;
const int mod = 1e9;
long long ans = 0;
int group[N];
vector<int> adj[N];
int sz[N];
int sub[N];
vector<int> v[N];
int n;
void dfs(int v,int par){
    sub[v] = sz[v];
    for(auto u:adj[v]){
        if(u == par)continue;
        //cout << v << " edge " << u << endl;
        dfs(u,v);
        sub[v] += sub[u];
        ans += (long long)sub[u]*(n-sub[u]);
    }
}
void solve(){
    int cnt = 1;
    int bef = 1;
    for(int i = 0;i<n;i++){
        int tmp = cnt;
        sort(v[i].begin(),v[i].end());
        for(int j = 0;j<v[i].size();j++){
            sz[cnt] = 0;
            int r = j-1;
            bool last = 0;
            while(r  + 1< v[i].size() && v[i][j] + (r-j+1) == v[i][r+1]){
                r++;
                //cout << v[i][r] << " ";
                if(last == 0 && bef <= group[v[i][r]]){
                    adj[group[v[i][r]]].push_back(cnt);
                    adj[cnt].push_back(group[v[i][r]]);
                    last = 1;
                }
                else last = 0;
                sz[cnt]++;
                group[v[i][r]] = cnt;
            }
            //cout << endl;
            j = r;
            cnt++;
        }
        bef = tmp;
    }
    dfs(1,0);
}
int DistanceSum(int n_, int *x, int *y) {
    n = n_;
    vector<int> compressx,compressy;
    for(int i = 0;i<n;i++){
        compressx.push_back(x[i]);
        compressy.push_back(y[i]);
    }
    sort(compressx.begin(),compressx.end());
    sort(compressy.begin(),compressy.end());
    compressx.resize(unique(compressx.begin(),compressx.end())-compressx.begin());
    compressy.resize(unique(compressy.begin(),compressy.end())-compressy.begin());
    map<pair<int,int>,int> mp;
    for(int i = 0;i<n;i++){
        x[i] = lower_bound(compressx.begin(),compressx.end(),x[i]) - compressx.begin();
        y[i] = lower_bound(compressy.begin(),compressy.end(),y[i]) - compressy.begin();
        mp[{x[i],y[i]}] = 1;
    }
    for(int i = 0;i<n;i++){
        v[x[i]].push_back(y[i]);
    }
    solve();
    for(int i = 0;i<n;i++){
        v[i].clear();
        adj[i].clear();
        group[i] = 0;
    }
    for(int i = 0;i<n;i++){
        v[y[i]].push_back(x[i]);
    }
    solve();
    return ans%mod;
}

Compilation message

city.cpp: In function 'void solve()':
city.cpp:28:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |         for(int j = 0;j<v[i].size();j++){
      |                       ~^~~~~~~~~~~~
city.cpp:32:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |             while(r  + 1< v[i].size() && v[i][j] + (r-j+1) == v[i][r+1]){
      |                   ~~~~~~^~~~~~~~~~~~~

Subtask #1
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 5012 KB Output is correct
2 Incorrect 3 ms 5008 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 21.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1078 ms 5108 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #3
0 / 23.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1081 ms 7124 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #4
0 / 45.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1084 ms 7312 KB Time limit exceeded
2 Halted 0 ms 0 KB -