1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
|
ull splitmix64(ull x) {
x += 0x9e3779b97f4a7c15;
x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
return x ^ (x >> 31);
}
struct custom_hash {
static const ull FIXED_RANDOM;
size_t operator()(ull x) const { return splitmix64(x + FIXED_RANDOM); }
};
const ull custom_hash::FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();
struct Info {
umap<int, int, custom_hash> ma;
int len = 0;
Info(int c = -1) {
if (c != -1) {
ma[c]++;
len = 1;
}
}
bool check(const vi& ma2) const {
int tem = 0;
rep(i, 0, sz(ma2) - 1) {
if (!ma.count(ma2[i])) continue;
tem += ma.find(ma2[i])->second;
}
return tem == len;
}
};
Info operator+(const Info& a, const Info& b) {
Info c;
c.len = a.len + b.len;
for (auto& [x, y] : a.ma) c.ma[x] += y;
for (auto& [x, y] : b.ma) c.ma[x] += y;
return c;
}
template <typename T>
class SegmentTree {
int n;
vector<T> tree;
T merge_val(T a, T b) const { return a + b; } // 合并子树
void maintain(int node) { // 维护整棵树
tree[node] = merge_val(tree[node * 2], tree[node * 2 + 1]);
}
void build(const vector<T>& a, int node, int l, int r) {
if (l == r) {
tree[node] = a[l];
return;
}
int m = (l + r) / 2;
build(a, node * 2, l, m);
build(a, node * 2 + 1, m + 1, r);
maintain(node);
} // 建树
void update(int node, int l, int r, int i, int val, int val2) {
tree[node].ma[val]--;
if (tree[node].ma[val] == 0) tree[node].ma.erase(tree[node].ma.find(val));
tree[node].ma[val2]++;
if (l == r) return;
int m = (l + r) / 2;
if (i <= m)
update(node * 2, l, m, i, val, val2);
else
update(node * 2 + 1, m + 1, r, i, val, val2);
} // 更新i处的值为val
T query(int node, int l, int r, int ql, int qr) const {
if (ql <= l && r <= qr) return tree[node];
int m = (l + r) / 2;
if (qr <= m) return query(node * 2, l, m, ql, qr);
if (ql > m) return query(node * 2 + 1, m + 1, r, ql, qr);
T l_res = query(node * 2, l, m, ql, qr);
T r_res = query(node * 2 + 1, m + 1, r, ql, qr);
return merge_val(l_res, r_res);
} // 查询[ql,qr]的值
int find_first(int node, int l, int r, int ql, int qr, const vi& val) const {
if (r < ql || l > qr) return -1;
if (ql <= l && r <= qr && tree[node].check(val)) return -1;
if (l == r) return l;
int m = (l + r) >> 1;
int res = find_first(node << 1, l, m, ql, qr, val);
if (res != -1) return res;
return find_first(node << 1 | 1, m + 1, r, ql, qr, val);
}
// 若固定左端点,需要记录前缀分段最大值,并加被待求区间完全覆盖的剪枝
int find_last(int node, int l, int r, int ql, int qr, const vi& val) const {
if (r < ql || l > qr) return -1;
if (ql <= l && r <= qr && tree[node].check(val)) return -1;
if (l == r) return l;
int m = (l + r) >> 1;
int res = find_last(node << 1 | 1, m + 1, r, ql, qr, val);
if (res != -1) return res;
return find_last(node << 1, l, m, ql, qr, val);
}
public:
SegmentTree(int n, T init_val) : SegmentTree(vector<T>(n, init_val)) {}
// 传入一个数组维护
SegmentTree(const vector<T>& a) : n(a.size()), tree(2 << bit_width(a.size() - 1)) { build(a, 1, 0, n - 1); }
void update(int i, int val, int val2) { update(1, 0, n - 1, i, val, val2); } // 更新i的值为val
T query(int ql, int qr) const { return query(1, 0, n - 1, ql, qr); } // 查询[ql,qr]的值
T get(int i) const { return query(1, 0, n - 1, i, i); } // 取出i处的值
// 查询[ql,qr]中第一个满足条件的下标
int find_first(int ql, int qr, const vi& val) const { return find_first(1, 0, n - 1, ql, qr, val); }
// 查询[ql,qr]中最后一个满足条件的下标
int find_last(int ql, int qr, const vi& val) const { return find_last(1, 0, n - 1, ql, qr, val); }
};
template <typename T = long long>
class Tree {
vector<T> tree;
public:
// 构造函数:初始化大小为 n 的树状数组,初始所有元素值为 val(外部表现为 0-based)
Tree(int n, T val = 0) : tree(n + 1) {
for (int i = 1; i <= n; i++) {
tree[i] += val;
int nxt = i + (i & -i);
if (nxt <= n) {
tree[nxt] += tree[i];
}
}
}
// 构造函数:使用给定的 vector 在 O(N) 时间内快速初始化建树
Tree(const vector<T>& data) {
int n = data.size();
tree.resize(n + 1);
for (int i = 1; i <= n; i++) {
tree[i] += data[i - 1]; // data是 0-based
int nxt = i + (i & -i);
if (nxt <= n) {
tree[nxt] += tree[i];
}
}
}
// 单点修改:将 0-based 下标 i 处的元素增加 val
void add(int i, T val = 1) {
for (++i; i < tree.size(); i += i & (-i)) {
tree[i] += val;
}
}
// 前缀求和:计算 0-based 下标区间 [0, i] 内的所有元素之和
T pre(int i) const {
T res = 0;
for (++i; i > 0; i &= i - 1) {
res += tree[i];
}
return res;
}
// 区间求和:计算 0-based 下标区间 [l, r] 内的所有元素之和
T query(int l, int r) const {
if (r < l) {
return 0;
}
return pre(r) - pre(l - 1); // 当 l=0 时, pre(-1) 会合理地返回 0
}
// 树上二分查找:返回满足前缀和 >= val 的最小 0-based 下标
int lower_bound(T val) const {
int w = bit_width(tree.size() - 1);
int res = 0;
T s = 0;
for (int i = w - 1; i >= 0; i--) {
int nxt = res + (1 << i);
if (nxt < tree.size() && tree[nxt] + s < val) {
res += (1 << i);
s += tree[nxt];
}
}
return res; // 返回 0-based 下标:内部 1-based 下标为 res + 1,因此 0-based 为 res
}
};
void solve() {
int n, q, op, x, y;
cin >> n >> q;
vi c(n);
rep(i, 0, n - 1) cin >> c[i];
vl v(n);
rep(i, 0, n - 1) cin >> v[i];
Tree tree(v);
vector<Info> init(n);
rep(i, 0, n - 1) init[i] = Info(c[i]);
SegmentTree<Info> tree2(init);
rep(i, 0, q - 1) {
cin >> op >> x;
x--;
if (op == 1) {
cin >> y;
tree2.update(x, c[x], y);
c[x] = y;
} else if (op == 2) {
cin >> y;
tree.add(x, y - v[x]);
v[x] = y;
} else {
cin >> y;
vi tem(y);
rep(j, 0, y - 1) cin >> tem[j];
int L = tree2.find_last(0, x - 1, tem);
int R = tree2.find_first(x + 1, n - 1, tem);
if (R == -1) R = n;
cout << tree.query(L + 1, R - 1) << endl;
}
}
return;
}
|