Submission #974548

#	Time	Username	Problem	Language	Result	Execution time	Memory
974548		rahidilbayramli	Arranging Shoes (IOI19_shoes)	C++17	10 / 100	75 ms	138060 KiB

shoes

#include "shoes.h"
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define ll long long
#define ld long double
#define vl vector<ll>
#define vi vector<int>
#define pii pair<int, int>
#define pll pair<ll, ll>
#define all(v) v.begin(), v.end()
#define pb push_back
#define f first
#define s second
using namespace std;
using namespace __gnu_pbds;
typedef tree<pll, null_type, less<pll>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;
const ll sz = 1e5+5;
stack<ll> st1[sz], st2[sz];
int segtree[4*sz], lazy[4*sz];
void push(ll v, ll l, ll r)
{
    if(lazy[v] != 0)
    {
        segtree[v] += (r - l + 1) * lazy[v];
        if(l != r)
        {
            lazy[2*v] += lazy[v];
            lazy[2*v+1] += lazy[v];
        }
        lazy[v] = 0;
    }
}
void update(ll v, ll l, ll r, ll tl, ll tr, ll val)
{
    push(v, l, r);
    if(l > r || l > tr || r < tl)
        return;
    if(tl <= l && r <= tr)
    {
        segtree[v] += (r - l + 1) * val;
        if(l != r)
        {
            lazy[2*v] += val;
            lazy[2*v+1] += val;
        }
        return;
    }
    else
    {
        ll mid = (l + r) / 2;
        update(2*v, l, mid, tl, min(mid, tr), val);
        update(2*v+1, mid+1, r, max(mid+1, tl), tr, val);
        segtree[v] = segtree[2*v] + segtree[2*v+1];
    }
}
ll findsum(ll v, ll l, ll r, ll tl, ll tr)
{
    if(l > r || l > tr || r < tl)
        return 0;
    push(v, l, r);
    if(tl <= l && r <= tr)
        return segtree[v];
    else
    {
        ll mid = (l + r) / 2;
        ll lans, rans;
        lans = findsum(2*v, l, mid, tl, min(mid, tr));
        rans = findsum(2*v+1, mid+1, r, max(mid+1, tl), tr);
        return lans + rans;
    }
}
long long count_swaps(vector<int> s) {
	ll ans = 0, i, n = s.size();
	for(i = 0; i < s.size(); i++)
    {
        if(s[i] > 0)
        {
            if(st1[s[i]].size() == 0)
                st2[s[i]].push(i);
            else
            {
                ll f = st1[s[i]].top();
                st1[s[i]].pop();
                update(1, 1, n, f+1, i-1, 1);
                ll sum = findsum(1, 1, n, f, f);
                ans += (i - f - 1);
                ans -= sum;
            }
        }
        else
        {
            if(st2[-s[i]].size() == 0)
                st1[-s[i]].push(i);
            else
            {
                ll f = st2[-s[i]].top();
                st2[-s[i]].pop();
                update(1, 1, n, f+1, i-1, 1);
                ll sum = findsum(1, 1, n, f, f);
                ans += (i - f);
                ans -= sum;
            }
        }
    }
    return ans;
}

Compilation message (stderr)

shoes.cpp: In function 'long long int count_swaps(std::vector<int>)':
shoes.cpp:75:15: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   75 |  for(i = 0; i < s.size(); i++)
      |             ~~^~~~~~~~~~

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