結果
| 問題 |
No.3251 kthmex
|
| ユーザー |
👑  hos.lyric
|
| 提出日時 | 2025-08-29 22:11:49 |
| 言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 6,722 bytes |
| コンパイル時間 | 2,126 ms |
| コンパイル使用メモリ | 174,368 KB |
| 実行使用メモリ | 25,600 KB |
| 最終ジャッジ日時 | 2025-08-29 22:12:34 |
| 合計ジャッジ時間 | 7,659 ms |
|
ジャッジサーバーID （参考情報） |
judge4 / judge1 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | AC * 4 TLE * 13 -- * 15 |
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:231:38: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
231 | for (int i = 0; i < N; ++i) scanf("%d", &A[i]);
| ~~~~~^~~~~~~~~~~~~
main.cpp:239:12: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
239 | scanf("%d%d%d", &O[q], &L[q], &R[q]);
| ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.cpp:242:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
242 | scanf("%d%d", &X[q], &Y[q]);
| ~~~~~^~~~~~~~~~~~~~~~~~~~~~
main.cpp:244:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
244 | scanf("%d", &K[q]);
| ~~~~~^~~~~~~~~~~~~
ソースコード
#include <cassert>
#include <cmath>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <bitset>
#include <chrono>
#include <complex>
#include <deque>
#include <functional>
#include <iostream>
#include <limits>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
using namespace std;
using Int = long long;
template <class T1, class T2> ostream &operator<<(ostream &os, const pair<T1, T2> &a) { return os << "(" << a.first << ", " << a.second << ")"; };
template <class T> ostream &operator<<(ostream &os, const vector<T> &as) { const int sz = as.size(); os << "["; for (int i = 0; i < sz; ++i) { if (i >= 256) { os << ", ..."; break; } if (i > 0) { os << ", "; } os << as[i]; } return os << "]"; }
template <class T> void pv(T a, T b) { for (T i = a; i != b; ++i) cerr << *i << " "; cerr << endl; }
template <class T> bool chmin(T &t, const T &f) { if (t > f) { t = f; return true; } return false; }
template <class T> bool chmax(T &t, const T &f) { if (t < f) { t = f; return true; } return false; }
#define COLOR(s) ("\x1b[" s "m")
// cannot use count
// no move constructor (==> use pointer for merge tech)
// unordered_set by value: __gnu_pbds::null_type
// no erase(iterator)
#include <ext/pb_ds/assoc_container.hpp>
using __gnu_pbds::gp_hash_table;
// https://codeforces.com/blog/entry/62393
#include <chrono>
struct Hash {
static uint64_t splitmix64(uint64_t x) {
// http://xorshift.di.unimi.it/splitmix64.c
x += 0x9e3779b97f4a7c15;
x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
return x ^ (x >> 31);
}
size_t operator()(uint64_t x) const {
static const uint64_t FIXED_RANDOM = std::chrono::steady_clock::now().time_since_epoch().count();
return splitmix64(x + FIXED_RANDOM);
}
size_t operator()(const pair<int, int> &a) const {
return operator()((uint64_t)a.first << 32 | a.second);
}
};
template <class K> using Set = gp_hash_table<K, __gnu_pbds::null_type, Hash>;
template <class K, class V> using Map = gp_hash_table<K, V, Hash>;
// iterator find_by_order(k): 0-indexed
// size_type order_of_key(key): min id s.t. >= key
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
template <class K, class V> using rbtree = __gnu_pbds::tree<
K,
V,
std::less<K>,
__gnu_pbds::rb_tree_tag,
__gnu_pbds::tree_order_statistics_node_update
>;
int N, Q;
vector<int> A;
vector<int> O, L, R, X, Y, K;
vector<int> brute() {
vector<int> ans(Q, 0);
auto as = A;
for (int q = 0; q < Q; ++q) {
if (O[q] == 1) {
for (int i = L[q]; i < R[q]; ++i) if (as[i] == X[q]) as[i] = Y[q];
} else {
vector<int> bs(as.begin() + L[q], as.begin() + R[q]);
sort(bs.begin(), bs.end());
bs.erase(unique(bs.begin(), bs.end()), bs.end());
int lo = 0, hi = (int)bs.size() + K[q];
for (; lo + 1 < hi; ) {
const int mid = lo + (hi - lo) / 2;
// ex in [1, mid]
int ex = mid;
ex -= (upper_bound(bs.begin(), bs.end(), mid) - bs.begin());
((ex >= K[q]) ? hi : lo) = mid;
}
ans[q] = hi;
}
}
return ans;
}
namespace fast {
constexpr int MAX_Y = 1010;
int U, V;
vector<int> as;
rbtree<int, int> freq[MAX_Y];
vector<pair<int, int>> lz[MAX_Y];
void add(int u, int a, int f) {
auto it = freq[u].find(a);
if (it != freq[u].end()) {
it->second += f;
if (!it->second) freq[u].erase(it);
} else {
freq[u][a] = f;
}
}
void push(int u) {
Map<int, int> ma;
for (; lz[u].size(); ) {
const int a = lz[u].back().first;
const int b = lz[u].back().second;
lz[u].pop_back();
int c = b;
auto it = ma.find(b);
if (it != ma.end()) {
c = it->second;
}
ma[a] = c;
}
const int l = u * V;
const int r = min((u + 1) * V, N);
for (int i = l; i < r; ++i) {
auto it = ma.find(as[i]);
if (it != ma.end()) {
as[i] = it->second;
}
}
}
vector<int> run() {
V = sqrt(N);
U = (N + V - 1) / V;
cerr<<"U = "<<U<<", V = "<<V<<endl;
as = A;
for (int u = 0; u < U; ++u) {
const int l = u * V;
const int r = min((u + 1) * V, N);
for (int i = l; i < r; ++i) {
add(u, as[i], +1);
}
lz[u].clear();
}
vector<int> ans(Q, 0);
for (int q = 0; q < Q; ++q) {
const int uL = L[q] / V;
const int uR = (R[q] - 1) / V;
if (uL == uR) {
push(uL);
} else {
push(uL);
push(uR);
}
if (O[q] == 1) {
auto modify = [&](int u, int l, int r) -> void {
int cnt = 0;
for (int i = l; i < r; ++i) if (as[i] == X[q]) {
as[i] = Y[q];
++cnt;
}
add(u, X[q], -cnt);
add(u, Y[q], +cnt);
};
if (uL == uR) {
modify(uL, L[q], R[q]);
} else {
modify(uL, L[q], (uL + 1) * V);
modify(uR, uR * V, R[q]);
for (int u = uL + 1; u < uR; ++u) {
auto it = freq[u].find(X[q]);
if (it != freq[u].end()) {
const int f = it->second;
freq[u].erase(it);
freq[u][Y[q]] += f;
lz[u].emplace_back(X[q], Y[q]);
}
}
}
} else {
auto calc = [&](int thr) -> int {
auto san = [&](int l, int r) -> int {
int cnt = 0;
for (int i = l; i < r; ++i) if (as[i] <= thr) ++cnt;
return cnt;
};
int ret = 0;
if (uL == uR) {
ret += san(L[q], R[q]);
} else {
ret += san(L[q], (uL + 1) * V);
ret += san(uR * V, R[q]);
for (int u = uL + 1; u < uR; ++u) {
ret += freq[u].order_of_key(thr + 1);
}
}
return ret;
};
int lo = 0, hi = (R[q] - L[q]) + K[q];
for (; lo + 1 < hi; ) {
const int mid = lo + (hi - lo) / 2;
((mid - calc(mid) >= K[q]) ? hi : lo) = mid;
}
ans[q] = hi;
}
}
return ans;
}
} // fast
int main() {
for (; ~scanf("%d%d", &N, &Q); ) {
A.resize(N);
for (int i = 0; i < N; ++i) scanf("%d", &A[i]);
O.resize(Q);
L.resize(Q);
R.resize(Q);
X.assign(Q, 0);
Y.assign(Q, 0);
K.assign(Q, 0);
for (int q = 0; q < Q; ++q) {
scanf("%d%d%d", &O[q], &L[q], &R[q]);
--L[q];
if (O[q] == 1) {
scanf("%d%d", &X[q], &Y[q]);
} else if (O[q] == 2) {
scanf("%d", &K[q]);
} else {
assert(false);
}
}
const auto ans = fast::run();
for (int q = 0; q < Q; ++q) if (O[q] == 2) {
printf("%d\n", ans[q]);
}
}
return 0;
}