The Preliminary Contest for ICPC Asia Xuzhou 2019 I. query (线段树+二维偏序)

题意:
给你一组 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[l1]再减去左端点在 [ 1 , l − 1 ] [1,l-1] [1,l1],右端点在 [ 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[l1]=x,剩下的是 y y y,求的就是 x − y x-y xy
对于一个数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[l1]就是 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[l1]就是 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;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值