题意:
给你一组
n
n
n个数字,这些数字是
1
n
1~n
1 n的一个排列。现在有很多个询问,询问给定一个区间,问区间内有多少个整除对。
思路:
设
s
u
m
[
i
]
sum[i]
sum[i]为从1到i的所有整除对数量。对于区间
[
l
,
r
]
[l,r]
[l,r],我们知道,区间
[
l
,
r
]
[l,r]
[l,r]的整除对数量是
s
u
m
[
r
]
−
s
u
m
[
l
−
1
]
sum[r]-sum[l-1]
sum[r]−sum[l−1]再减去左端点在
[
1
,
l
−
1
]
[1,l-1]
[1,l−1],右端点在
[
l
,
r
]
[l,r]
[l,r]内的整除对,所以可以离线做。设
s
u
m
[
r
]
−
s
u
m
[
l
−
1
]
=
x
sum[r]-sum[l-1]=x
sum[r]−sum[l−1]=x,剩下的是
y
y
y,求的就是
x
−
y
x-y
x−y。
对于一个数i,在区间
[
1
,
n
]
[1,n]
[1,n]内有
n
/
i
n/i
n/i个与i组成整除对的数。可以利用这个性质来计算
y
y
y,按顺序枚举
i
i
i,对于区间
[
l
,
r
]
[l,r]
[l,r]的询问,当
l
=
i
l=i
l=i时,
s
u
m
[
r
]
−
s
u
m
[
l
−
1
]
sum[r]-sum[l-1]
sum[r]−sum[l−1]就是
y
y
y的值。然后再把
s
u
m
sum
sum内
i
i
i的倍数加1,因为i也应该算在
x
x
x内,不在
y
y
y内。对于
x
x
x,则当
r
=
i
r=i
r=i时,
s
u
m
[
r
]
−
s
u
m
[
l
−
1
]
sum[r]-sum[l-1]
sum[r]−sum[l−1]就是
x
x
x。
用线段树维护
s
u
m
sum
sum,复杂度可以优化到
n
l
o
g
2
n
nlog_{2}^{n}
nlog2n。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5 + 10;
int a[N], n, m, t[N << 2], pos[N], ans[N];
struct node {
int l, r, id;
} q1[N], q2[N];
void add(int l, int r, int rt, int x) {
if(l == r) {
t[rt]++;
return;
}
int m = l + r >> 1;
if(x <= m)
add(l, m, rt << 1, x);
else
add(m + 1, r, rt << 1 | 1, x);
t[rt] = t[rt << 1] + t[rt << 1 | 1];
}
int query(int l, int r, int rt, int L, int R) {
if(L <= l && r <= R)
return t[rt];
int m = l + r >> 1, ans = 0;
if(L <= m)
ans += query(l, m, rt << 1, L, R);
if(R > m)
ans += query(m + 1, r, rt << 1 | 1, L, R);
return ans;
}
int main() {
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
pos[a[i]] = i;
}
for(int i = 1; i <= m; i++) {
scanf("%d%d", &q1[i].l, &q1[i].r);
q1[i].id = i;
q2[i] = q1[i];
}
sort(q1 + 1, q1 + m + 1, [](const node & a, const node & b) {
return a.l == b.l ? a.r < b.r : a.l < b.l;
});
sort(q2 + 1, q2 + m + 1, [](const node & a, const node & b) {
return a.r == b.r ? a.l < b.l : a.r < b.r;
});
for(int i = 1, j = 1, k = 1; i <= n; i++) {
while(q1[j].l == i && j <= m) {
ans[q1[j].id] -= query(1, n, 1, q1[j].l, q1[j].r);
j++;
}
for(int s = 2; s * a[i] <= n; s++)
add(1, n, 1, pos[s * a[i]]);
while(q2[k].r == i && k <= m) {
ans[q2[k].id] += query(1, n, 1, q2[k].l, q2[k].r);
k++;
}
}
for(int i = 1; i <= m; i++)
printf("%d\n", ans[i]);
return 0;
}