Codeforces 301D Yaroslav and Divisors 【树状数组 + 思维】

60 篇文章 0 订阅
27 篇文章 0 订阅

题目链接:Codeforces 301D Yaroslav and Divisors

题意:给你 n 个数和m次查询,每次询问区间 [l,r] 里面有多少对 (p,q) 满足 a[p] a[q] 整除。 (l<=p,q<=r)

思路:T到死了,弱B各种优化,最后给神奇的思路跪了。
我们考虑区间 [l,r] 的方案数,记 x=[1,r] 的方案数 - [1,l] 的方案数, y=p 在区间 [1,l1] q 在区间[l,r]的方案数。
ans[l,r]=xy
TMD,菜鸡根本没想过这样写,一直想着用树状数组不断维护区间端点 [l,r] 。然后就死循环了。。。
维护左端点(按 l 升序sort)——在处理到 [l,r] 的时候,我们先去掉方案数 y
维护右端点(按r升序 sort )——在处理到 [l,r] 的时候,我们直接统计 x 即可,因为y已经减过了。

AC 代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#define PI acos(-1.0)
#define CLR(a, b) memset(a, (b), sizeof(a))
#define fi first
#define se second
#define ll o<<1
#define rr o<<1|1
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
const int MAXN = 2*1e5 + 1;
const int pN = 1e6;// <= 10^7
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
void add(LL &x, LL y) { x += y; x %= MOD; }
int C[MAXN];
int n, m;
int lowbit(int x) {
    return x & (-x);
}
void add(int x, int d) {
    while(x <= n) {
        C[x] += d;
        x += lowbit(x);
    }
}
int Sum(int x) {
    int s = 0;
    while(x > 0) {
        s += C[x];
        x -= lowbit(x);
    }
    return s;
}
struct Node {
    int l, r, id;
};
Node num[MAXN], num1[MAXN];
bool cmp(Node a, Node b) {
    return a.l != b.l ? a.l < b.l : a.r < b.r;
}
bool cmp1(Node a, Node b) {
    return a.r != b.r ? a.r < b.r : a.l < b.l;
}
int ans[MAXN];
int a[MAXN], pos[MAXN];
void Update(int x) {
    for(int i = x; i <= n; i += x) {
        add(pos[i], 1);
    }
}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        pos[a[i]] = i;
    }
    CLR(C, 0);
    for(int i = 1; i <= m; i++) {
        scanf("%d%d", &num[i].l, &num[i].r);
        num[i].id = i;
        num1[i].l = num[i].l;
        num1[i].r = num[i].r;
        num1[i].id = i;
    }
    sort(num+1, num+m+1, cmp);
    sort(num1+1, num1+m+1, cmp1);
    for(int i = 1; i <= m; i++) ans[i] = 0;
    int j = 1, k = 1;
    for(int i = 1; i <= n; i++) {
        while(j <= m && num[j].l == i) {
            ans[num[j].id] -= Sum(num[j].r) - Sum(num[j].l-1);
            j++;
        }
        Update(a[i]);
        while(k <= m && num1[k].r == i) {
            ans[num1[k].id] += Sum(num1[k].r) - Sum(num1[k].l-1);
            k++;
        }
    }
    for(int i = 1; i <= m; i++) {
        printf("%d\n", ans[i]);
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值