제출 #594586

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
594586		davi_bart	정렬하기 (IOI15_sorting)	C++14	56 / 100	1093 ms	8000 KiB

sorting

#pragma GCC optimize("O3")
#include <bits/stdc++.h>

#include "sorting.h"
using namespace std;
#define ll long long
// #define int ll
#define fi first
#define se second
#define ld long double
#define pb push_back
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int N;
struct Dsu {
    vector<int> par, dim;
    Dsu(vector<int> &k) {
        par.resize(N);
        dim = vector<int>(N, 1);
        iota(par.begin(), par.end(), 0);
        for (int i = 0; i < k.size(); i++) {
            unite(i, k[i]);
        }
    }
    int find(int pos) {
        return par[pos] = par[pos] == pos ? pos : find(par[pos]);
    }
    void unite(int a, int b) {
        a = find(a);
        b = find(b);
        if (a == b) return;
        if (dim[a] < dim[b]) swap(a, b);
        par[b] = a;
        dim[a] += dim[b];
    }
    int conta() {
        int tot = 0;
        for (int i = 0; i < N; i++) {
            if (i == par[i]) tot += dim[i] - 1;
        }
        return tot;
    }
};
vector<int> v;
vector<int> ini;
int findSwapPairs(int N, int S[], int M, int P[], int Q[], int X[], int Y[]) {
    ::N = N;
    for (int i = 0; i < N; i++) v.pb(S[i]);
    ini = v;
    int ans = 0;
    for (int i = 0; i < M; i++) {
        Dsu dsu(v);
        // cout << dsu.conta() << " ";
        if (dsu.conta() <= i) {
            ans = i;
            break;
        }
        swap(v[P[i]], v[Q[i]]);
    }
    // cout << endl;
    // int o = 0;
    vector<pair<int, int> > scambi;
    while (1) {
        int k = scambi.size();
        for (int i = 0; i < N; i++) {
            if (v[i] != i) {
                scambi.pb({v[i], i});
                swap(v[find(v.begin(), v.end(), i) - v.begin()], v[i]);
            }
        }
        if (k == scambi.size()) break;
    }
    v = ini;
    for (int i = 0; i < ans; i++) {
        swap(v[P[i]], v[Q[i]]);
        for (int j = 0; j < N; j++) {
            if (v[j] == scambi[i].fi) X[i] = j;
        }
        for (int j = 0; j < N; j++) {
            if (v[j] == scambi[i].se) Y[i] = j;
        }
        swap(v[X[i]], v[Y[i]]);
    }
    // v = ini;
    // for (int i = 0; i < ans; i++) {
    //     swap(v[P[i]], v[Q[i]]);
    //     swap(v[X[i]], v[Y[i]]);
    // }
    // for (int i = 0; i < N; i++) assert(v[i] == i);
    return ans;
    v = ini;
    multiset<int> ok;
    vector<bool> valid;
    for (int i = 0; i < ans; i++) {
        Dsu dsu(v);
        if (dsu.find(P[i]) == dsu.find(Q[i])) {
            swap(v[P[i]], v[Q[i]]);
            ok.insert(P[i]);
            ok.insert(Q[i]);
            valid.pb(1);
        } else
            valid.pb(0);
        cout << valid[i] << endl;
    }
    for (int i = 0; i < ans; i++) {
        if (valid[i]) {
            swap(v[P[i]], v[Q[i]]);
            ok.erase(ok.find(P[i]));
            ok.erase(ok.find(Q[i]));
            for (int j = 0; j < N; j++) {
                if (v[j] != j && ok.count(j) == 0) {
                    X[i] = j;
                    Y[i] = v[j];
                    swap(v[j], v[v[j]]);
                    break;
                }
            }
        } else {
            X[i] = P[i];
            Y[i] = Q[i];
        }
    }
    return ans;
}

컴파일 시 표준 에러 (stderr) 메시지

sorting.cpp: In constructor 'Dsu::Dsu(std::vector<int>&)':
sorting.cpp:20:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 |         for (int i = 0; i < k.size(); i++) {
      |                         ~~^~~~~~~~~~
sorting.cpp: In function 'int findSwapPairs(int, int*, int, int*, int*, int*, int*)':
sorting.cpp:45:23: warning: declaration of 'N' shadows a global declaration [-Wshadow]
   45 | int findSwapPairs(int N, int S[], int M, int P[], int Q[], int X[], int Y[]) {
      |                   ~~~~^
sorting.cpp:13:5: note: shadowed declaration is here
   13 | int N;
      |     ^
sorting.cpp:63:28: warning: conversion from 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   63 |         int k = scambi.size();
      |                 ~~~~~~~~~~~^~
sorting.cpp:70:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   70 |         if (k == scambi.size()) break;
      |             ~~^~~~~~~~~~~~~~~~

• Subtask 1: 8 / 8
• Subtask 2: 12 / 12
• Subtask 3: 16 / 16
• Subtask 4: 0 / 18
• Subtask 5: 20 / 20
• Subtask 6: 0 / 26