제출 #1105801

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1105801		snpmrnhlol	Arranging Shoes (IOI19_shoes)	C++17	100 / 100	48 ms	15756 KiB

shoes

#include "shoes.h"
#include <iostream>
using namespace std;
typedef long long ll;
const int N = 1e5;
int fen[2*N];
vector <int> pos[N];
vector <int> pos2[N];
int pt[N*2];
int n;
void add(int pos, int x){
    for(int i = pos;i < n;i|=(i + 1)){
        fen[i]+=x;
    }
}
int get(int pos){
    int r = 0;
    for(int i = pos;i >= 0;i&=(i + 1),i--){
        r+=fen[i];
    }
    return r;
}
long long count_swaps(std::vector<int> s){
    n = s.size();
    for(int i = 0;i < n;i++){
        if(s[i] > 0){
            pt[i] = (int)pos[s[i] - 1].size();
            pos[s[i] - 1].push_back(i);
        }else{
            pt[i] = (int)pos2[-s[i] - 1].size();
            pos2[-s[i] - 1].push_back(i);
        }
        add(i, 1);
    }
    ll ans = 0;
    for(int i = 0;i < n;i++){
        int id1 = pos[(s[i] > 0?s[i] - 1:-s[i] - 1)][pt[i]];
        int id2 = pos2[(s[i] > 0?s[i] - 1:-s[i] - 1)][pt[i]];
        //cout<<id1<<' '<<id2<<'\n';
        if(id2 < id1)swap(id2,id1);
        if(id2 == i)continue;
        add(id1, -1);
        add(id2, -1);
        ans+=get(id2);
        if(s[id1] > 0)ans++;
    }
	return ans;
}

• Subtask 1: 10 / 10
• Subtask 2: 20 / 20
• Subtask 3: 20 / 20
• Subtask 4: 15 / 15
• Subtask 5: 20 / 20
• Subtask 6: 15 / 15