Submission #1177016

#	Time	Username	Problem	Language	Result	Execution time	Memory
1177016		Kaztaev_Alisher	Nile (IOI24_nile)	C++20	32 / 100	71 ms	14408 KiB

nile

#include <bits/stdc++.h>

#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define all(a) a.begin() , a.end()
#define F first
#define S second

using namespace std;
using ll = long long;

const ll N = 2e5+5 , inf = 2e9 + 7;
const ll INF = 1e18 ,   mod = 1e9+7;

ll w[N] , a[N] , b[N] , w1[N] , ord[N] , d[N] , n , q;
ll p[N] , sz[N];
ll cnt;
int get(int x){
	if(p[x] == x) return x;
	return p[x] = get(p[x]);
}
void merge(int a, int b){
	a = get(a);
	b = get(b);
	if(sz[b] % 2) cnt--;
	if(sz[a] % 2) cnt--;
	sz[a] += sz[b];
	p[b] = a;
	if(sz[a] % 2) cnt++;
}
vector<long long> solve(){
	vector<ll> vec(q,0);
	sort(w+1,w+1+n);
	cnt = n;
	for(int i = 1; i <= n; i++){
		p[i] = i; sz[i] = 1;
	}
	for(int i = 1; i <= q; i++) ord[i] = i;
	sort(ord+1,ord+1+q,[&](int i , int j){
		return d[i] < d[j];
	});
	sort(p+1,p+1+q);
	multiset<pair<int,int>> st;
	for(int i = 1; i+1 <= n; i++){
		st.insert({abs(w[i]-w[i+1]) , i});
	}
	for(int _ = 1; _ <= q; _++){
		int i = ord[_];
		while(st.size() && st.begin()->F <= d[i]){
			pair<ll,ll> x = *st.begin();
			st.erase(st.find(x));
			int a = x.S;
			int b = x.S+1;
			merge(a,b);
		}
		vec[i-1] = cnt+n;
	}
	return vec;
}
vector<long long> calculate_costs(vector<int> W, vector<int> A, vector<int> B, vector<int> E){
	n = W.size();
	for(int i = 1; i <= n; i++){
		w[i] = W[i-1];
		a[i] = A[i-1];
		b[i] = B[i-1];
	}
	q = E.size();
	for(int i = 1; i <= q; i++){
		d[i] = E[i-1];
	}
	return solve();
}

// int main() {
  // int N;
  // assert(1 == scanf("%d", &N));
  // std::vector<int> W(N), A(N), B(N);
  // for (int i = 0; i < N; i++)
    // assert(3 == scanf("%d%d%d", &W[i], &A[i], &B[i]));
  // int Q;
  // assert(1 == scanf("%d", &Q));
  // std::vector<int> E(Q);
  // for (int j = 0; j < Q; j++)
    // assert(1 == scanf("%d", &E[j]));
  // fclose(stdin);
// 
  // std::vector<long long> R = calculate_costs(W, A, B, E);
// 
  // int S = (int)R.size();
  // for (int j = 0; j < S; j++)
    // printf("%lld\n", R[j]);
  // fclose(stdout);
// 
  // return 0;
// }

• Subtask 0: 0 / 0
• Subtask 1: 0 / 6
• Subtask 2: 0 / 13
• Subtask 3: 17 / 17
• Subtask 4: 0 / 11
• Subtask 5: 0 / 20
• Subtask 6: 15 / 15
• Subtask 7: 0 / 18