結果

問題 No.2809 Sort Query
コンテスト
ユーザー kwm_t
提出日時 2024-07-12 23:12:56
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 5,699 bytes
コンパイル時間 3,515 ms
コンパイル使用メモリ 261,604 KB
実行使用メモリ 176,064 KB
最終ジャッジ日時 2024-07-12 23:13:39
合計ジャッジ時間 36,808 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 6 WA * 65
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
//#include <atcoder/all>
using namespace std;
//using namespace atcoder;
//using mint = modint1000000007;
//const int mod = 1000000007;
//using mint = modint998244353;
//const int mod = 998244353;
//const int INF = 1e9;
//const long long LINF = 1e18;
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep2(i,l,r)for(int i=(l);i<(r);++i)
#define rrep(i, n) for (int i = (n) - 1; i >= 0; --i)
#define rrep2(i,l,r)for(int i=(r) - 1;i>=(l);--i)
#define all(x) (x).begin(),(x).end()
#define allR(x) (x).rbegin(),(x).rend()
#define P pair<int,int>
template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }
template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }
#include <array>
#include <vector>
struct BinaryTrie {
	int kind;
	std::vector<int>node;
	std::vector<std::array<int, 2>>next;
	std::array<int, 2> e{ -1,-1 };
	BinaryTrie(int _kind) :kind(_kind) {
		node.resize(1);
		next.emplace_back(e);
	}
	void insert(const std::vector<int>&v) {
		int now = 0;
		for (int i = 0; i < v.size(); ++i) {
			if (-1 == next[now][v[i]]) {
				next[now][v[i]] = node.size();
				node.emplace_back(0);
				next.emplace_back(e);
			}
			now = next[now][v[i]];
			node[now]++;
		}
	}
	// 1-indexed
	std::vector<int>kth_element(int k) {
		int now = 0;
		std::vector<int>ret;
		while (true) {
			bool flag = false;
			for (int i = 0; i < kind; ++i) {
				if (-1 == next[now][i])continue;
				if (k <= node[next[now][i]]) {
					flag = true;
					ret.emplace_back(i);
					now = next[now][i];
					break;
				}
				else {
					k -= node[next[now][i]];
				}
			}
			if (!flag)break;
		}
		return ret;
	}
	bool seach(const std::vector<int>&v) {
		int now = 0;
		for (int i = 0; i < (int)v.size(); ++i) {
			if (-1 == next[now][v[i]])return false;
			if (node[next[now][v[i]]] <= 0)return false;
			now = next[now][v[i]];
		}
		int count = node[now];
		for (int i = 0; i < kind; ++i)if (-1 != next[now][i])count -= node[next[now][i]];
		return count;
	}
	void erase(const std::vector<int>&v) {
		int now = 0;
		for (int i = 0; i < (int)v.size(); ++i) {
			now = next[now][v[i]];
			node[now]--;
		}
	}
	// Binary Trie
	std::vector<int>xor_max(const std::vector<int>&v) {
		std::vector<int>ret;
		int now = 0;
		for (int i = 0; i < (int)v.size(); ++i) {
			std::vector<int>tmp = { 1 - v[i],v[i] };
			for (int j = 0; j < (int)tmp.size(); ++j) {
				if (-1 == next[now][tmp[j]])continue;
				if (node[next[now][tmp[j]]] <= 0)continue;
				ret.emplace_back(tmp[j]);
				now = next[now][tmp[j]];
				break;
			}
		}
		return ret;
	}
	std::vector<int>xor_min(const std::vector<int>&v) {
		std::vector<int>ret;
		int now = 0;
		for (int i = 0; i < (int)v.size(); ++i) {
			std::vector<int>tmp = { v[i],1 - v[i] };
			for (int j = 0; j < (int)tmp.size(); ++j) {
				if (-1 == next[now][tmp[j]])continue;
				if (node[next[now][tmp[j]]] <= 0)continue;
				ret.emplace_back(tmp[j]);
				now = next[now][tmp[j]];
				break;
			}
		}
		return ret;
	}
	// xとxorした結果がk未満になる物の数(0含む)
	int xor_count(const std::vector<int>&x, const std::vector<int>&k) {
		int ret = 0, now = 0;
		for (int i = 0; i < (int)k.size(); ++i) {
			if (0 == k[i]) {
				if (-1 == next[now][x[i]])break;
				now = next[now][x[i]];
			}
			else {
				if (-1 != next[now][x[i]])ret += node[next[now][x[i]]];
				if (-1 == next[now][1 - x[i]])break;
				now = next[now][1 - x[i]];
			}
		}
		return ret;
	}
	// xより大きいものの個数
	int over_count(const std::vector<int>&x) {
		int ret = 0, now = 0;
		for (int i = 0; i < (int)x.size(); ++i) {
			if (0 == x[i]) {
				if (-1 != next[now][1])ret += node[next[now][1]];
			}
			if (-1 == next[now][x[i]])break;
			now = next[now][x[i]];
		}
		return ret;
	}
	std::vector<int> trans(long long n, int sz = 60) {
		std::vector<int>ret(sz, 0);
		for (int i = 0; i < sz; ++i) {
			if (1 & (n >> i))ret[sz - 1 - i] = 1;
		}
		return ret;
	}
	long long inverse(const std::vector<int>&v) {
		long long ret = 0;
		for (int i = 0; i < v.size(); ++i) {
			ret *= 2;
			ret += v[i];
		}
		return ret;
	}
};
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n, q; cin >> n >> q;
	vector<long long>a(n);
	rep(i, n)cin >> a[i];
	struct Query {
		int t;
		int k;
		long long x;
		Query(int t = -1, int k = -1, long long x = -1) :t(t), k(k), x(x) {}
		void output() {
			std::cout << t << " " << k << " " << x << endl;
		}
	};
	vector<Query>query(q);
	rep(i, q) {
		int t, k; 
		long long x;
		cin >> t;
		if (1 == t)cin >> k >> x, k--;
		else if (3 == t)cin >> k, k--;
		query[i] = Query(t, k, x);
	}
	//rep(i, q) {
	//	cout << query[i].k << " " << query[i].t << " " << query[i].x << endl;
	//}
	int i = 0;
	for (; i < q;) {
		while (i < q && 2 != query[i].t) {
			if (3 == query[i].t) cout << a[query[i].k] << endl;
			else a[query[i].k] = query[i].x;
			i++;
		}
		break;
	}
	BinaryTrie bt(2);
	rep(i, n)bt.insert(bt.trans(a[i]));
	i++;
	return 0;

	for (; i < q;) {
		map<int, long long>mp;//index,value
		while (i < q && 2 != query[i].t) {
			//query[i].output();
			int index = query[i].k;
			if (3 == query[i].t) {
				if (mp.count(index)) {
					cout << mp[index] << endl;
				}
				else {
					auto v = bt.kth_element(index + 1);
					cout << bt.inverse(v) << endl;
				}
			}
			else {
				mp[index] = query[i].x;
			}
			i++;
		}
		vector<pair<long long, long long>>vec;
		for (auto[k, v] : mp) {
			auto val = bt.kth_element(k + 1);
			vec.push_back({ bt.inverse(val) ,v });
		}
		for (auto p : vec) {
			bt.erase(bt.trans(p.first));
			bt.insert(bt.trans(p.second));
		}
		// sort
	}
	return 0;
}
0