码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces 475D CGCDSSQ 区间gcd值

时间:2018-01-27 18:57:54      阅读:100      评论:0      收藏:0      [点我收藏+]

标签:2-2   需要   bre   def   typedef   name   gpo   因此   test   

题目链接

题意

给定一个长度为 \(n\) 的数列 \(a_1,...,a_n\)\(q\) 个询问 \(x_1,...,x_q\),对于每个 \(x_i\) 回答有多少对 \((l,r)\) 满足\(\ (1\leq l\leq r\leq n)\)\(gcd(a_l,a_{l+1},...,a_r)=x_i\)

思路

对于固定的右端点 \(i\),将左端点从 (\(i\)) 向 (\(1\)) 延伸,\(gcd\) 值是递减的,且变化次数不超过 \(logC\) (\(C\)为数列中最大值)

下面讲述两种方法,第一种效率高一些,而第二种也提供了一些新的见解。

法一:滚动数组 —— 更新分段信息

枚举右端点,将由左端点划分出的 \(gcd\) 值分段。每次用新加进来的 \(a_i\) 去与刚刚的若干段再取 \(gcd\) 并更新分段信息,更新的同时统计数目。

保存与更新分段信息 可用滚动数组实现,统计数目 则显然用map(要注意的一点是:需要用map<int, LL>,因为数目可能会爆\(int\))。

法二:二分 + ST表 —— 找gcd值变化位置

参考自 hzwer.

如果说上一种做法是极大程度地利用了 上一次的信息,那么这一种做法就是抓住了 gcd值具有单调性 这个特点。

因此,确定分段位置可以直接采用二分查找,而如何快速地获取某一段的 \(gcd\) 值呢?就靠 \(ST\)大显身手了。

// 学到两点:
// 1. ST表适用的范围不仅局限于区间极值问题
// 2. 系统自带的log是真的慢...

Code

Ver. 1 : 171ms

#include <bits/stdc++.h>
#define maxn 100010
using namespace std;
typedef long long LL;
int a[maxn];
int gcd(int a, int b) { return b ? gcd(b, a % b) : a; }
struct node { int x, p; };
map<int, LL> mp;
vector<node> v[2];
int main() {
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; ++i) scanf("%d", &a[i]);
    for (int i = 0; i < n; ++i) {
        bool me = i & 1,
            op = !me;
        v[me].clear();
        v[me].push_back({a[i], i});
        int last = a[i];
        for (auto nd : v[op]) {
            int temp = gcd(nd.x, a[i]);
            if (temp == last) v[me][v[me].size()-1].p = nd.p;
            else v[me].push_back({temp, nd.p}), last = temp;
        }
        int now = i;
        for (auto nd : v[me]) {
            int pre = nd.p;
            mp[nd.x] += now - pre + 1;
            now = pre - 1;
        }
    }
    int q, x;
    scanf("%d", &q);
    while (q--) {
        scanf("%d", &x);
        printf("%I64d\n", mp[x]);
    }
    return 0;
}

Ver. 2 : 296ms

#include <bits/stdc++.h>
#define maxn 100010
using namespace std;
typedef long long LL;
int gcd[maxn][32], a[maxn], n, Log[maxn], bin[32];
map<int, LL> mp;
int Gcd(int a, int b) { return b ? Gcd(b, a%b) : a; }
void rmqInit() {
    Log[0] = -1; bin[0] = 1;
    for (int i = 1; i < 20; ++i) bin[i] = bin[i-1] << 1;
    for (int i = 1; i <= n; ++i) Log[i] = Log[i>>1] + 1, gcd[i][0] = a[i];
    for (int j = 1; bin[j] <= n; ++j) {
        for (int i = 1; i + bin[j-1] - 1 <= n; ++i) {
            gcd[i][j] = Gcd(gcd[i][j-1], gcd[i + bin[j-1]][j-1]);
        }
    }
}
int query(int l, int r) {
    int k = Log[r-l+1];
    return Gcd(gcd[l][k], gcd[r-bin[k]+1][k]);
}
int bi(int i, int l, int r, int x) {
    while (r-l>1) {
        int mid = l+r >> 1, val = query(i, mid);
        if (val >= x) l = mid;
        else r = mid - 1;
    }
    return query(i, r) == x ? r : l;
}
int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
    rmqInit();
    for (int i = 1; i <= n; ++i) {
        int l = i;
        while (true) {
            if (l == n+1) break;
            int val = query(i, l);
            int r = bi(i, l, n, val);
            mp[val] += r-l+1;
            l = r+1;
        }
    }
    int q, x;
    scanf("%d", &q);
    while (q--) {
        scanf("%d", &x);
        printf("%I64d\n", mp[x]);
    }
    return 0;
}

Codeforces 475D CGCDSSQ 区间gcd值

标签:2-2   需要   bre   def   typedef   name   gpo   因此   test   

原文地址:https://www.cnblogs.com/kkkkahlua/p/8366339.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!