Submission #681279

#	Time	Username	Problem	Language	Result	Execution time	Memory
681279		whqkrtk04	Arranging Shoes (IOI19_shoes)	C++17	45 / 100	72 ms	14856 KiB

shoes

#include "shoes.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<int, pii> piii;
typedef pair<ll, ll> pll;
typedef pair<ll, pll> plll;
#define fi first
#define se second
const int INF = 1e9+1;
const int P = 1000000007;
const ll LLINF = (ll)1e18+1;
template <typename T>
ostream& operator<<(ostream& os, const vector<T>& v) { for(auto i : v) os << i << " "; os << "\n"; return os; }
template <typename T1, typename T2>
ostream& operator<<(ostream& os, const pair<T1, T2>& p) { os << p.fi << " " << p.se; return os; }

class segtree {
	private:
	int n;
	vector<int> seg;
	void update(int i, int s, int e, int j, int x) {
		if(j >= e || s > j) return;
		if(s+1 == e) seg[i] += x;
		else {
			update(i<<1, s, s+e>>1, j, x);
			update(i<<1|1, s+e>>1, e, j, x);
			seg[i] = seg[i<<1]+seg[i<<1|1];
		}
	}
	int query(int i, int s, int e, int l, int r) {
		if(e <= l || r <= s) return 0;
		if(l <= s && e <= r) return seg[i];
		return query(i<<1, s, s+e>>1, l, r)+query(i<<1|1, s+e>>1, e, l, r);
	}
	public:
	segtree(int n) {
		this->n = n;
		seg.resize(4*n);
	}
	void update(int j) { update(1, 0, n, j, 1); }
	int query(int j) { return query(1, 0, n, j, n); }
};

ll count_swaps(vector<int> s) {
	vector<vector<int>> A(s.size());
	for(int i = 0, cnt = 0; i < s.size(); i++) {
		if(s[i] > 0) {
			A[s[i]-1].push_back(cnt);
			s[i] = 2*(cnt++)+1;
		}
	}
	vector<int> C(s.size(), 0);
	for(int i = 0; i < s.size(); i++) {
		if(s[i] < 0) {
			int idx = -s[i]-1;
			s[i] = 2*A[idx][C[idx]++];
		}
	}
	segtree S(s.size());
	ll ans = 0LL;
	for(int i = 0; i < s.size(); i++) {
		ans += S.query(s[i]);
		S.update(s[i]);
	}
	return ans;
}

Compilation message (stderr)

shoes.cpp: In member function 'void segtree::update(int, int, int, int, int)':
shoes.cpp:27:21: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   27 |    update(i<<1, s, s+e>>1, j, x);
      |                    ~^~
shoes.cpp:28:20: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   28 |    update(i<<1|1, s+e>>1, e, j, x);
      |                   ~^~
shoes.cpp: In member function 'int segtree::query(int, int, int, int, int)':
shoes.cpp:35:26: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   35 |   return query(i<<1, s, s+e>>1, l, r)+query(i<<1|1, s+e>>1, e, l, r);
      |                         ~^~
shoes.cpp:35:54: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   35 |   return query(i<<1, s, s+e>>1, l, r)+query(i<<1|1, s+e>>1, e, l, r);
      |                                                     ~^~
shoes.cpp: In function 'll count_swaps(std::vector<int>)':
shoes.cpp:48:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |  for(int i = 0, cnt = 0; i < s.size(); i++) {
      |                          ~~^~~~~~~~~~
shoes.cpp:55:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |  for(int i = 0; i < s.size(); i++) {
      |                 ~~^~~~~~~~~~
shoes.cpp:63:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |  for(int i = 0; i < s.size(); i++) {
      |                 ~~^~~~~~~~~~

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