Submission #1241943

#	Time	Username	Problem	Language	Result	Execution time	Memory
1241943		shiori_chan	즐거운 사진 수집 (JOI13_collecting)	C++17	100 / 100	1153 ms	98912 KiB

collecting

#include <iostream>
#include <vector>
#include <random>
#include <algorithm>
#include <chrono>
#define all(x) x.begin(), x.end()
#define vi vector <int> 

using namespace std;
typedef long long ll;
#define int long long

const int nd = 1048576 + 1;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
uniform_int_distribution<int> dist(1 , (int)2e9);

vector <vi> segtree(2 , vi(4 * nd)) , count_node(2 , vi(22 , 0));
int timer = 1e9;
int base = 0;
bool ar = 0;

void build(int id , int l , int r , int layer , vi &cnt , vi &st){
	if(l == r){
		st[id] = 0;
		return;
	}
	int mid = (r + l) / 2;
	build(id * 2 , l , mid , layer + 1 , cnt , st); 
	build(id * 2 + 1 , mid + 1 , r , layer + 1 , cnt , st);

	st[id] = (++ timer) * (st[id * 2] != st[id * 2 + 1]);
	++ cnt[layer];
}

void up(int id , int l , int r , int x , int layer , vi &cnt, vi &st){
	if(l > x || r < x) return;
	if(l == r){
		st[id] ^= 1;
		return;
	}

	int mid = (r + l) / 2;
	up(id * 2 , l , mid , x , layer + 1 , cnt , st); 
	up(id * 2 + 1 , mid + 1 , r , x , layer + 1 , cnt , st);
	
	int pre = st[id];
	if(st[id * 2] == st[id * 2 + 1]) st[id] = st[id * 2];
	else st[id] = ++ timer;

	if(pre < 2 && st[id] > 2) -- cnt[layer];
	else if(pre > 2 && st[id] < 2) ++ cnt[layer];
}

void solve(){
	int n , q; cin >> n >> q;
	int num_layer = n;
	n = (1 << n);
	int all = 0;
	for(int i = 0;i < num_layer; ++ i) all += (1ll << i) * (1ll << i);
	base = all;

	build(1 , 1 , n , 0 , count_node[0] , segtree[0]);
	build(1 , 1 , n , 0 , count_node[1] ,segtree[1]);

	while(q --){
		int t , x; cin >> t >> x;
		ar = t ^ 1;
		up(1 , 1 , n , x , 0 , count_node[t] , segtree[t]);
		base = 0;
		for(int i = 0 ;i < num_layer; ++ i) base += count_node[0][i] * count_node[1][i];

		cout << (all - base) * 4 + 1 << '\n'; 
	}	
}

signed main() {
   ios_base::sync_with_stdio(false);
   cin.tie(0);

	#define task "task" 
	if(fopen(task".inp", "r")) {
		freopen(task".inp", "r", stdin);
		freopen(task".out", "w", stdout);
	}
	solve();

	return 0;
}

Compilation message (stderr)

collecting.cpp: In function 'int main()':
collecting.cpp:82:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   82 |                 freopen(task".inp", "r", stdin);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
collecting.cpp:83:24: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   83 |                 freopen(task".out", "w", stdout);
      |                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 20 / 20
• Subtask 3: 70 / 70