제출 #571649

#제출 시각아이디문제언어결과실행 시간메모리
571649Sergio_2357Arranging Shoes (IOI19_shoes)C++17
45 / 100
455 ms39252 KiB
#include "shoes.h"
#include <bits/stdc++.h>
 
using namespace std;
 
#define int long long
 
typedef vector<int> vi;
typedef vector<signed> vs;
typedef vector<vi> vii;
 
const int sizea = 200000;
const int sqrts = 316;
const int sqrtn = (sizea / sqrts + 2);
 
vi norms(sizea+sqrts);
vi sqrtarr(sqrtn, 0);
 
void update(int b, int e, int v)
{
    int idx = sizea;
    for (int i = b; i < e; i++) {
        norms[i] += v;
        if (i % sqrts == sqrts - 1) {
            idx = i + sqrts;
            break;
        }
    }
    for (int i = idx; i < e; i += sqrts) {
        sqrtarr[i / sqrts] += v;
        idx = i + 1;
    }
    for (int i = idx; i < e; i++) {
        norms[i] += v;
    }
}
 
int get(int i)
{
    return norms[i] + sqrtarr[i / sqrts];
}
 
int gidx(int v, int n)
{
    return v + (n / 2);
}
 
long long count_swaps(vs s_s)
{
    vi s(s_s.size());
    int n = s.size();
    for (int i = 0; i < n; i++)
        s[i] = s_s[i];
    iota(norms.begin(), norms.end(), 0);
    vi used(n, 0);
    map<int, vector<int>> last;
    set<int> idx;
    for (int i = 0; i < n; i++)
        idx.insert(s[i]);
    for (int i = 0; i < n; i++)
        last[s[i]].push_back(i);
    for (int i : idx)
        sort(last[i].begin(), last[i].end());
    for (int i : idx)
        reverse(last[i].begin(), last[i].end());
    long long c = 0;
    for (int i = 0; i < n; i++) {
        if (used[i])
            continue;
        //cout << i << endl;
        int r = -s[i];
        int nxt = *prev(last[r].end());
        last[s[i]].pop_back();
        last[r].pop_back();
        used[i] = 1;
        used[nxt] = 1;
        int d = get(nxt) - get(i);
        d--;
        if (s[i] > 0)
            d++;
        c += d;
        update(i + 1, nxt, 1);
    }
    return c;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...