제출 #312812

#제출 시각아이디문제언어결과실행 시간메모리
312812BasilhijazArranging Shoes (IOI19_shoes)C++17
50 / 100
552 ms544916 KiB
#include "shoes.h"
#include <bits/stdc++.h>

using namespace std;

#define pb push_back

vector<pair<int, int> > pairs;

const int N = 1e6;
int seg[4*N];
bool done[N];


void build(int p, int s, int e){
    if(s == e){
        seg[p] = 1;
        return;
    }
    build(2*p, s, (s + e)/2);
    build(2*p + 1, (s + e)/2 + 1, e);
    seg[p] = seg[2*p] + seg[2*p + 1];
}

void update(int p, int s, int e, int i){
    if(s == e){
        if(s == i){
            seg[p] = 0;
        }
        return;
    }
    if(i <= (s + e)/2)update(2*p, s, (s + e)/2, i);
    else update(2*p + 1, (s + e)/2 + 1, e, i);
    seg[p] = seg[2*p] + seg[2*p + 1];
}

int get(int p, int s, int e, int a, int b){
    if(a <= s && b >= e){
        return seg[p];
    }
    else if(b < s || a > e){
        return 0;
    }
    return get(2 * p, s, (s + e)/2, a, b) + get(2 * p + 1, (s + e)/2 + 1, e, a, b);
}
long long count_swaps(vector<int> s) {
    int n = s.size();
    queue<int> left[2*n + 1];
    queue<int> right[2*n + 1];
    for(int i = 0; i < n; i++){
        if(s[i] < 0){
            if(!right[abs(s[i])].empty()){
                pairs.pb({right[abs(s[i])].front(), i});
                right[abs(s[i])].pop();
            }
            else{
                left[abs(s[i])].push(i);
            }
        }
        else{
            if(!left[s[i]].empty()){
                pairs.pb({left[s[i]].front(), i});
                left[s[i]].pop();
            }
            else{
                right[s[i]].push(i);
            }
        }
    }
    build(1, 0, n - 1);
    int ans = 0;
    sort(pairs.begin(), pairs.end());
    for(int i = 0; i < pairs.size(); i++){
        if(s[pairs[i].first] > 0){
            ans++;
        }
    }
    memset(done, 0, sizeof(done));
    for(int i = 0; i < pairs.size(); i++){
        update(1, 0, n - 1, pairs[i].second);
        ans += get(1, 0, n - 1, pairs[i].first + 1, pairs[i].second - 1);
    }
	return ans;
}

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

shoes.cpp: In function 'long long int count_swaps(std::vector<int>)':
shoes.cpp:73:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   73 |     for(int i = 0; i < pairs.size(); i++){
      |                    ~~^~~~~~~~~~~~~~
shoes.cpp:79:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   79 |     for(int i = 0; i < pairs.size(); i++){
      |                    ~~^~~~~~~~~~~~~~
#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...