LuoguP1972 - [SDOI2009] HH的项链
给定若干区间,询问区间内物品的种数。
考虑离线,将每一个区间按照右端点从小到大进行排序。右端点不断向右移动的时候,将移动时遇到的序列中的数获取过来,更新我们已有的信息,然后进行统计。我们要掌握的就是每一种物品离右端点最近的位置在哪,统计最近的位置,就是为了在固定右端点的情况下,区间再小,也能掌握到这一种物品的存在。
以题目样例为例, [ 1 , 2 , 3 , 4 , 3 , 5 ] [1,2,3,4,3,5] [1,2,3,4,3,5],我们设一个答案序列 [ 0 , 0 , 0 , 0 , 0 , 0 ] [0,0,0,0,0,0] [0,0,0,0,0,0],用来方便我们对于每一个询问的统计。设右端点为 r r r,样例中的物品序列为 a a a。
当 r = 1 r=1 r=1时, a 1 = 1 a_1=1 a1=1, 1 1 1是之前没有遇到过的数,将这个位置标上 1 1 1。答案序列变为 [ 1 , 0 , 0 , 0 , 0 , 0 ] [1,0,0,0,0,0] [1,0,0,0,0,0]。
当 r = 2 r=2 r=2时, a 2 = 2 a_2=2 a2=2, 2 2 2是之前没有遇到过的数,将这个位置标上 1 1 1。答案序列变为 [ 1 , 1 , 0 , 0 , 0 , 0 ] [1,1,0,0,0,0] [1,1,0,0,0,0]。
此时有一个询问是 [ 1 , 2 ] [1,2] [1,2],那么就求答案序列 [ 1 , 2 ] [1,2] [1,2]的和,也就是 2 2 2。
继续移动右端点。
当 r = 3 r=3 r=3时, a 3 = 3 a_3=3 a3=3, 3 3 3是之前没有遇到过的数,将这个位置标上 1 1 1。答案序列变为 [ 1 , 1 , 1 , 0 , 0 , 0 ] [1,1,1,0,0,0] [1,1,1,0,0,0]。
当 r = 4 r=4 r=4时, a 4 = 4 a_4=4 a4=4, 4 4 4是之前没有遇到过的数,将这个位置标上 1 1 1。答案序列变为 [ 1 , 1 , 1 , 1 , 0 , 0 ] [1,1,1,1,0,0] [1,1,1,1,0,0]。
当 r = 5 r=5 r=5时, a 5 = 3 a_5=3 a5=3,与 a 3 a_3 a3相同,将原先值为 3 3 3上的 1 1 1抹除,将这个位置标上 1 1 1。答案序列变为 [ 1 , 1 , 0 , 1 , 1 , 0 ] [1,1,0,1,1,0] [1,1,0,1,1,0]。
此时有一个询问是 [ 3 , 5 ] [3,5] [3,5],那么就求答案序列 [ 3 , 5 ] [3,5] [3,5]的和,也就是 2 2 2。
继续移动右端点。
当 r = 6 r=6 r=6时, a 6 = 5 a_6=5 a6=5, 5 5 5是之前没有遇到过的数,将这个位置标上 1 1 1。答案序列变为 [ 1 , 1 , 0 , 1 , 1 , 1 ] [1,1,0,1,1,1] [1,1,0,1,1,1]。
此时有一个询问是 [ 2 , 6 ] [2,6] [2,6],那么就求答案序列 [ 2 , 6 ] [2,6] [2,6]的和,也就是 4 4 4。
总结起来,就是将一种物品的代表设置得离右端点越近越好,这样可以保证,左端点以左的标记在询问区间内一定没有对应的相同物品。
求答案序列的和,用线段树/树状数组都可以。更新答案序列用普通数组记录一下这个数组上一次出现的位置即可。
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1000005;
int n, m;
int a[N], b[N], c[N];
int lowbit(int x) {
return x & (-x);
}
void add(int x, int k) {
while (x <= n) {
c[x] += k; x += lowbit(x);
}
}
int presum(int x) {
int ans = 0;
while (x >= 1) {
ans += c[x]; x -= lowbit(x);
}
return ans;
}
struct QUERY {
int l, r, ans, id;
}q[N];
int mp[N];
void main2() {
cin >> n;
for (int i = 1; i <= n; ++i) {
cin >> a[i];
}
cin >> m;
for (int i = 1; i <= m; ++i) {
cin >> q[i].l >> q[i].r;
q[i].ans = 0; q[i].id = i;
}
sort(q + 1, q + m + 1, [](const QUERY &A, const QUERY &B) {
return A.r < B.r;
});
int lst = 0;
for (int i = 1; i <= m; ++i) {
if (q[i].r > lst) {
for (int j = lst + 1; j <= q[i].r; ++j) {
if (!mp[a[j]]) {
mp[a[j]] = j;
add(j, 1);
}
else {
int x = mp[a[j]];
add(x, -1);
add(j, 1);
mp[a[j]] = j;
}
}
lst = q[i].r;
}
q[i].ans = presum(q[i].r) - presum(q[i].l - 1);
}
sort(q + 1, q + m + 1, [](const QUERY &A, const QUERY &B) {
return A.id < B.id;
});
for (int i = 1; i <= m; ++i) {
cout << q[i].ans << '\n';
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
LL _;
// cin >> _;
_ = 1;
while (_--) main2();
return 0;
}
LuoguP4113 - [HEOI2012]采花
给定若干区间,询问区间内出现过两次及以上的物品的种数。
和上面那道题大同小异,只有几行代码的差别。这一道题要同时记录这种花上一次出现的位置和上上一次出现的位置。右端点向右移动时,每遇到一种花,就要在这种话上一次出现的位置上打一个标记,如果这是他三次及以上出现,就要把上上一次出现的位置取消掉。
以样例中的数据为例,设序列 a a a为题目花的种类数据 [ 1 , 2 , 2 , 3 , 1 ] [1,2,2,3,1] [1,2,2,3,1]。同样设答案序列 [ 0 , 0 , 0 , 0 , 0 ] [0,0,0,0,0] [0,0,0,0,0]。
r
=
1
r=1
r=1,
a
1
=
1
a_1=1
a1=1,这是第一次遇到
1
1
1,所以只记录出现的位置,不在答案序列进行标记。
此时答案序列
[
0
,
0
,
0
,
0
,
0
]
[0,0,0,0,0]
[0,0,0,0,0]。
r
=
2
r=2
r=2,
a
2
=
2
a_2=2
a2=2,这是第一次遇到
2
2
2,所以只记录出现的位置,不在答案序列进行标记。
此时答案序列
[
0
,
0
,
0
,
0
,
0
]
[0,0,0,0,0]
[0,0,0,0,0]。
此时有以 2 2 2为右端点的询问,分别是 [ 1 , 2 ] [1,2] [1,2]和 [ 2 , 2 ] [2,2] [2,2]。分别求得答案序列中 [ 1 , 2 ] [1,2] [1,2]和 [ 2 , 2 ] [2,2] [2,2]的区间和,答案都是 0 0 0。
r
=
3
r=3
r=3,
a
3
=
2
a_3=2
a3=2,这是第二次遇到
2
2
2,所以在答案序列对
2
2
2上一次出现的位置进行标记,并记录先在出现的位置。
此时答案序列
[
0
,
1
,
0
,
0
,
0
]
[0,1,0,0,0]
[0,1,0,0,0]。
此时有以 3 3 3为右端点的询问,是 [ 2 , 3 ] [2,3] [2,3]。分别求得答案序列中 [ 2 , 3 [2,3 [2,3的区间和,答案是 1 1 1。
r
=
4
r=4
r=4,
a
4
=
3
a_4=3
a4=3,这是第一次遇到
3
3
3,所以只记录出现的位置,不在答案序列进行标记。
此时答案序列
[
0
,
1
,
0
,
0
,
0
]
[0,1,0,0,0]
[0,1,0,0,0]。
r
=
5
r=5
r=5,
a
5
=
1
a_5=1
a5=1,这是第二次遇到
1
1
1,所以在答案序列对
1
1
1上一次出现的位置进行标记,并记录先在出现的位置。
此时答案序列
[
1
,
1
,
0
,
0
,
0
]
[1,1,0,0,0]
[1,1,0,0,0]。
此时有以 5 5 5为右端点的询问,分别是 [ 1 , 5 ] [1,5] [1,5]和 [ 3 , 5 ] [3,5] [3,5]。分别求得答案序列中 [ 1 , 5 ] [1,5] [1,5]和 [ 3 , 5 ] [3,5] [3,5]的区间和,答案分别是 2 2 2和 0 0 0。
这个题数据范围很大,线段树常数比较大,线段树不太容易过,但是树状数组是可以过的。
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 2000005;
int n, co, m;
int a[N], b[N], c[N];
int lowbit(int x) {
return x & (-x);
}
void add(int x, int k) {
while (x <= n) {
c[x] += k; x += lowbit(x);
}
}
int presum(int x) {
int ans = 0;
while (x >= 1) {
ans += c[x]; x -= lowbit(x);
}
return ans;
}
struct QUERY {
int l, r, ans, id;
}q[N];
int mp[N], del[N];
void main2() {
cin >> n >> co >> m;
for (int i = 1; i <= co; ++i) {
mp[i] = del[i] = 0;
}
for (int i = 1; i <= n; ++i) {
cin >> a[i];
}
for (int i = 1; i <= m; ++i) {
cin >> q[i].l >> q[i].r;
q[i].ans = 0; q[i].id = i;
}
sort(q + 1, q + m + 1, [](const QUERY &A, const QUERY &B) {
return A.r < B.r;
});
int lst = 0;
for (int i = 1; i <= m; ++i) {
if (q[i].r > lst) {
for (int j = lst + 1; j <= q[i].r; ++j) {
if (!mp[a[j]]) {
mp[a[j]] = j;
}
else {
int x = 0;
if (del[a[j]]) x = del[a[j]];
if (x) {
add(x, -1);
}
x = mp[a[j]];
add(x, 1);
del[a[j]] = x;
mp[a[j]] = j;
}
}
lst = q[i].r;
}
q[i].ans = presum(q[i].r) - presum(q[i].l - 1);
}
sort(q + 1, q + m + 1, [](const QUERY &A, const QUERY &B) {
return A.id < B.id;
});
for (int i = 1; i <= m; ++i) {
cout << q[i].ans << '\n';
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
LL _;
// cin >> _;
_ = 1;
while (_--) main2();
return 0;
}
(最后一个点1.86s,也超级悬!)
代码源469 - Closest Equals
考虑离线来做这道题。将所有的询问按照 r r r排序。我们考察 r r r不断向右的时候,每向右遇到一个数,如何维护答案。
以样例 1 1 1为例, [ 1 , 1 , 2 , 3 , 2 ] [1,1,2,3,2] [1,1,2,3,2]。我们最初设答案是 [ i n f , i n f , i n f , i n f , i n f ] [inf,inf,inf,inf,inf] [inf,inf,inf,inf,inf]。
r = 1 r=1 r=1时, a i = 1 a_i=1 ai=1,前面没有与其相等的数,所以不会更新答案。
r = 2 r=2 r=2时, a i = 1 a_i=1 ai=1,最近一个与其相等的数是 a 1 a_1 a1。那么让答案序列变为 [ 2 − 1 , i n f , i n f , i n f , i n f ] = [ 1 , i n f , i n f , i n f , i n f ] [2-1,inf,inf,inf,inf]=[1,inf,inf,inf,inf] [2−1,inf,inf,inf,inf]=[1,inf,inf,inf,inf]。这样的话,在以 2 2 2为右端点的查询区间中,只要区间包含 1 1 1,那么选取答案最小值就可以选到最优答案 1 1 1。
r = 3 r=3 r=3时, a i = 2 a_i=2 ai=2,前面没有与其相等的数,所以不会更新答案。
r = 4 r=4 r=4时, a i = 3 a_i=3 ai=3,前面没有与其相等的数,所以不会更新答案。
r = 5 r=5 r=5时, a i = 2 a_i=2 ai=2,最近一个与其相等的数是 a 3 a_3 a3。那么让答案序列变为 [ 1 , 2 , 2 , 2 , i n f ] [1,2,2,2,inf] [1,2,2,2,inf]。对每一个位置, 2 = 5 − 3 2=5-3 2=5−3,是 a 3 a_3 a3与 a 5 a_5 a5的距离。我们对答案序列的每一个位置都是更新最小值,如果当前位置的答案已经小于要更新上去的答案,那就不动。
总体上来说,维护这样一个答案序列,当 r r r移到序列第 i i i个数时,每遇到一个前面遇到的一个数,假设这个数与相同的数的最近距离是 x x x,那么更新 [ 1 , i − 1 ] [1,i-1] [1,i−1],将这些位置的答案与 x x x比较,选取更小的值。查询答案时,就选取答案序列中 [ l , r ] [l,r] [l,r]中的最小值作为答案。
维护这样的序列相当于区间赋值,维护最小值,用线段树可以轻松解决。
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
struct QUERY {
int l, r, ans, id;
}q[500005];
int a[500005];
int n, m;
struct Tree {
int l, r, tag, mn;
}t[2000005];
void pd(int ni) {
t[ni << 1].tag = min(t[ni << 1].tag, t[ni].tag);
t[ni << 1 | 1].tag = min(t[ni << 1 | 1].tag, t[ni].tag);
t[ni << 1].mn = min(t[ni << 1].mn, t[ni].tag);
t[ni << 1 | 1].mn = min(t[ni << 1 | 1].mn, t[ni].tag);
t[ni].tag = 1e9;
}
void pu(int ni) {
t[ni].mn = min(t[ni << 1].mn, t[ni << 1 | 1].mn);
}
void build_tree(int ni, int l, int r) {
t[ni].l = l; t[ni].r = r; t[ni].tag = 1e9;
if (l == r) {
t[ni].mn = 1e9;
return;
}
int mid = (l + r) >> 1;
build_tree(ni << 1, l, mid);
build_tree(ni << 1 | 1, mid + 1, r);
pu(ni);
}
void fix(int ni, int l, int r, int x) {
if (l <= t[ni].l and t[ni].r <= r) {
t[ni].tag = min(t[ni].tag, x);
t[ni].mn = min(t[ni].mn, t[ni].tag);
return;
}
int mid = (t[ni].l + t[ni].r) >> 1;
pd(ni);
if (l <= mid) fix(ni << 1, l, r, x);
if (mid < r) fix(ni << 1 | 1, l, r, x);
pu(ni);
}
int query(int ni, int l, int r) {
if (l <= t[ni].l and t[ni].r <= r) {
return t[ni].mn;
}
int mid = (t[ni].l + t[ni].r) >> 1;
pd(ni);
int ans = 1e9;
if (l <= mid) ans = min(ans, query(ni << 1, l, r));
if (mid < r) ans = min(ans, query(ni << 1 | 1, l, r));
return ans;
}
map<int, int> mp;
void main2() {
cin >> n >> m;
for (int i = 1; i <= n; ++i) {
cin >> a[i];
}
for (int i = 1; i <= m; ++i) {
cin >> q[i].l >> q[i].r;
q[i].ans = 0; q[i].id = i;
}
sort(q + 1, q + m + 1, [](const QUERY &A, const QUERY &B) {
return A.r < B.r;
});
int lst = 0;
build_tree(1, 1, n);
for (int i = 1; i <= m; ++i) {
if (q[i].r > lst) {
for (int j = lst + 1; j <= q[i].r; ++j) {
if (!mp[a[j]]) mp[a[j]] = j;
else {
int x = mp[a[j]];
fix(1, 1, x, j - x);
mp[a[j]] = j;
}
}
lst = q[i].r;
}
int res = query(1, q[i].l, q[i].r);
if (res == 1e9) q[i].ans = -1;
else q[i].ans = res;
}
sort(q + 1, q + m + 1, [](const QUERY &A, const QUERY &B) {
return A.id < B.id;
});
for (int i = 1; i <= m; ++i) {
cout << q[i].ans << '\n';
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
LL _;
// cin >> _;
_ = 1;
while (_--) main2();
return 0;
}