oj.uz

Submission #388250

# Submission time Handle Problem Language Result Execution time Memory
388250 2021-04-10T16:21:55 Z ACmachine Sorting (IOI15_sorting) C++17
100 / 100
 299 ms 27828 KB 
#include "sorting.h"
#include <bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;

template<typename T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
template<typename T, typename U> using ordered_map = tree<T, U, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

typedef long long ll;
typedef long double ld;

typedef pair<int,int> pii;
typedef pair<ll,ll> pll;

typedef vector<int> vi;
typedef vector<ll> vll;

#define FOR(i,j,k,in) for(int i=(j); i < (k);i+=in)
#define FORD(i,j,k,in) for(int i=(j); i >=(k);i-=in)
#define REP(i,b) FOR(i,0,b,1)
#define REPD(i,b) FORD(i,b,0,1)
#define pb push_back
#define mp make_pair
#define ff first
#define ss second
#define all(x) begin(x), end(x)
#define MANY_TESTS int tcase; cin >> tcase; while(tcase--)

const double EPS = 1e-9;
const int MOD = 1e9+7;
const ll INFF = 1e18;
const int INF = 1e9;
const ld PI = acos((ld)-1);
const vi dy = {1, 0, -1, 0, -1, 1, 1, -1};
const vi dx = {0, 1, 0, -1, -1, 1, -1, 1};

void DBG(){cout << "]" << endl;}
template<typename T, typename ...U> void DBG(const T& head, const U... args){ cout << head << "; "; DBG(args...); }
#define dbg(...) cout << "Line(" << __LINE__ << ") -> [" << #__VA_ARGS__ << "]: [", DBG(__VA_ARGS__);
#define chk() cout << "Check at line(" << __LINE__ << ") hit." << endl;

template<class T, unsigned int U>
ostream& operator<<(ostream& out, const array<T, U> &v){out << "[";  REP(i, U) out << v[i] << ", ";  out << "]"; return out;}
template <class T, class U>
ostream& operator<<(ostream& out, const pair<T, U> &par) {out << "[" << par.first << ";" << par.second << "]"; return out;}
template <class T>
ostream& operator<<(ostream& out, const set<T> &cont) { out << "{"; for( const auto &x:cont) out << x << ", "; out << "}"; return out; }
template <class T, class U>
ostream& operator<<(ostream& out, const map<T, U> &cont) {out << "{"; for( const auto &x:cont) out << x << ", "; out << "}"; return out; }
template<class T>
ostream& operator<<(ostream& out, const vector<T> &v){ out << "[";  REP(i, v.size()) out << v[i] << ", ";  out << "]"; return out;}

template<class T>
istream& operator>>(istream& in, vector<T> &v){ for(auto &x : v) in >> x; return in; }

int trivial_sort(int n, vector<int> &positions, vector<int>& P, vector<int> &Q){
    int id = 0;
    REP(i, n){
        REP(j, n){
            if(positions[j] == i){
                if(j != i){
                    swap(positions[i], positions[j]);
                    P[id] = positions[i]; Q[id] = positions[j];
                    id++;
                    break;
                }
            }
        }
    }
    return id;
}
int fast_sort(int n, vector<int> &positions, vector<int> &P, vector<int> &Q){ // fill P, Q with values swapped; positions is initial array
    vector<int> pos(n);
    auto swap2 = [&](int a, int b){
        swap(positions[a], positions[b]);
        pos[positions[a]] = a;
        pos[positions[b]] = b;
    };
    REP(i, positions.size()){
        pos[positions[i]] = i;
    }
    int id = 0;
    REP(i, n){
        int f = pos[i];
        if(f == i) continue;
        P[id] = positions[f]; Q[id] = positions[i];
        swap2(f, i);
        id++;
    }
    return id;
}
void adapt(vector<int> &init, vector<int> &X, vector<int> &Y, vector<int> &P, vector<int> &Q){
    vector<int> pos(init.size());
    REP(i, init.size()){
        pos[init[i]] = i;
    }
    auto swap2 = [&](int a, int b){
        swap(init[a], init[b]);
        pos[init[a]] = a;
        pos[init[b]] = b;
    };
    REP(i, X.size()){
        swap2(X[i], Y[i]);
        P[i] = pos[P[i]];
        Q[i] = pos[Q[i]];
        swap2(P[i], Q[i]);
    }
}
int findSwapPairs(int n, int S[], int m, int X[], int Y[], int P[], int Q[]) {
    vector<int> nX(m);
    vector<int> nY(m);
    REP(i, m){
        nX[i] = X[i]; nY[i] = Y[i];
    }
	int ans = INF;
	int l = 0;
	int r = m - 1;
	vector<int> init_arr(n);
	vector<int> arr(n);
	REP(i, n) init_arr[i] = S[i];
	vector<int> nP(m);
	vector<int> nQ(m);

	while(l <= r){
	    fill(all(nP), 0);
	    fill(all(nQ), 0);
	    arr = init_arr;
        int mid = (l + r) >> 1;
        REP(i, mid){
            swap(arr[X[i]], arr[Y[i]]);
        }
        int num = fast_sort(n, arr, nP, nQ);
        if(num <= mid){
            ans = mid;
            r = mid - 1;
        }
        else{
            l = mid + 1;
        }
	}

	//ans = 3;
    fill(all(nP), 0);
    fill(all(nQ), 0);
    arr = init_arr;
    REP(i, ans){
        swap(arr[X[i]], arr[Y[i]]);
    }
	fast_sort(n, arr, nP, nQ);
	adapt(init_arr, nX, nY, nP, nQ);
	REP(i, ans) {P[i] = nP[i]; Q[i] = nQ[i];}
	return ans;
}

Compilation message

sorting.cpp: In function 'int fast_sort(int, std::vector<int>&, std::vector<int>&, std::vector<int>&)':
sorting.cpp:20:40: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 | #define FOR(i,j,k,in) for(int i=(j); i < (k);i+=in)
      |                                        ^
sorting.cpp:22:18: note: in expansion of macro 'FOR'
   22 | #define REP(i,b) FOR(i,0,b,1)
      |                  ^~~
sorting.cpp:81:5: note: in expansion of macro 'REP'
   81 |     REP(i, positions.size()){
      |     ^~~
sorting.cpp: In function 'void adapt(std::vector<int>&, std::vector<int>&, std::vector<int>&, std::vector<int>&, std::vector<int>&)':
sorting.cpp:20:40: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 | #define FOR(i,j,k,in) for(int i=(j); i < (k);i+=in)
      |                                        ^
sorting.cpp:22:18: note: in expansion of macro 'FOR'
   22 | #define REP(i,b) FOR(i,0,b,1)
      |                  ^~~
sorting.cpp:96:5: note: in expansion of macro 'REP'
   96 |     REP(i, init.size()){
      |     ^~~
sorting.cpp:20:40: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 | #define FOR(i,j,k,in) for(int i=(j); i < (k);i+=in)
      |                                        ^
sorting.cpp:22:18: note: in expansion of macro 'FOR'
   22 | #define REP(i,b) FOR(i,0,b,1)
      |                  ^~~
sorting.cpp:104:5: note: in expansion of macro 'REP'
  104 |     REP(i, X.size()){
      |     ^~~

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 204 KB Output is correct
2 Correct 0 ms 204 KB Output is correct
3 Correct 0 ms 204 KB Output is correct
4 Correct 0 ms 204 KB Output is correct
5 Correct 0 ms 204 KB Output is correct
6 Correct 0 ms 204 KB Output is correct
7 Correct 0 ms 204 KB Output is correct

Subtask #2
12.0 / 12.0

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

Subtask #3
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 204 KB Output is correct
2 Correct 0 ms 204 KB Output is correct
3 Correct 1 ms 332 KB Output is correct
4 Correct 1 ms 332 KB Output is correct
5 Correct 1 ms 332 KB Output is correct
6 Correct 0 ms 204 KB Output is correct

Subtask #4
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 204 KB Output is correct
2 Correct 0 ms 204 KB Output is correct
3 Correct 0 ms 204 KB Output is correct
4 Correct 0 ms 204 KB Output is correct
5 Correct 0 ms 204 KB Output is correct
6 Correct 0 ms 204 KB Output is correct
7 Correct 0 ms 204 KB Output is correct
8 Correct 1 ms 204 KB Output is correct
9 Correct 0 ms 204 KB Output is correct
10 Correct 1 ms 332 KB Output is correct
11 Correct 1 ms 332 KB Output is correct
12 Correct 1 ms 332 KB Output is correct
13 Correct 0 ms 204 KB Output is correct
14 Correct 0 ms 204 KB Output is correct
15 Correct 1 ms 332 KB Output is correct
16 Correct 1 ms 332 KB Output is correct
17 Correct 1 ms 332 KB Output is correct
18 Correct 0 ms 204 KB Output is correct
19 Correct 1 ms 204 KB Output is correct
20 Correct 0 ms 204 KB Output is correct
21 Correct 2 ms 588 KB Output is correct
22 Correct 2 ms 588 KB Output is correct
23 Correct 2 ms 588 KB Output is correct
24 Correct 2 ms 588 KB Output is correct
25 Correct 2 ms 588 KB Output is correct
26 Correct 2 ms 588 KB Output is correct
27 Correct 2 ms 588 KB Output is correct

Subtask #5
20.0 / 20.0

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

Subtask #6
26.0 / 26.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 460 KB Output is correct
2 Correct 2 ms 460 KB Output is correct
3 Correct 2 ms 460 KB Output is correct
4 Correct 1 ms 460 KB Output is correct
5 Correct 1 ms 460 KB Output is correct
6 Correct 2 ms 460 KB Output is correct
7 Correct 2 ms 460 KB Output is correct
8 Correct 3 ms 460 KB Output is correct
9 Correct 2 ms 460 KB Output is correct
10 Correct 3 ms 460 KB Output is correct
11 Correct 2 ms 460 KB Output is correct
12 Correct 2 ms 460 KB Output is correct
13 Correct 2 ms 460 KB Output is correct
14 Correct 1 ms 460 KB Output is correct
15 Correct 224 ms 16840 KB Output is correct
16 Correct 241 ms 25212 KB Output is correct
17 Correct 267 ms 26396 KB Output is correct
18 Correct 128 ms 24240 KB Output is correct
19 Correct 206 ms 26400 KB Output is correct
20 Correct 228 ms 27068 KB Output is correct
21 Correct 220 ms 27056 KB Output is correct
22 Correct 219 ms 21872 KB Output is correct
23 Correct 252 ms 27456 KB Output is correct
24 Correct 270 ms 27084 KB Output is correct
25 Correct 264 ms 26680 KB Output is correct
26 Correct 220 ms 27056 KB Output is correct
27 Correct 199 ms 26484 KB Output is correct
28 Correct 259 ms 27100 KB Output is correct
29 Correct 252 ms 26712 KB Output is correct
30 Correct 170 ms 26284 KB Output is correct
31 Correct 299 ms 27324 KB Output is correct
32 Correct 248 ms 26584 KB Output is correct
33 Correct 261 ms 27084 KB Output is correct
34 Correct 260 ms 26960 KB Output is correct
35 Correct 208 ms 25936 KB Output is correct
36 Correct 109 ms 25924 KB Output is correct
37 Correct 264 ms 27828 KB Output is correct
38 Correct 260 ms 26676 KB Output is correct
39 Correct 263 ms 26832 KB Output is correct