HDU5869 Different GCD Subarray Query (区间gcd 离线查询 + 线段树/树状数组)

题目链接: Different GCD Subarray Query

大致题意

有一个长度为n的序列, 从1~n编号, 每个位置都有一个元素值.

多次询问, 每次询问[l, r]区间, 在该区间内所有子数组能产生多少种不同的gcd.

解题思路

离线查询 + 树状数组/线段树

相似题目推荐

首先理解一下题意, 说白了就是每次询问有两层循环, i从l->r, j从i->r, 对于每一个子数组[i, j], 把所有元素求gcd, 放到一个set里. 最后问你set里有多少个元素. (所以读完发现了一个 n³还带log的算法.)

首先这个题我们要考虑到gcd的一个性质.
假设现在有n个数字, 从1~n编号, 我们从左到右逐个开始gcd, 当到第i个数字时, 有两种情况.
①gcd不变, 即: [1, i - 1]的gcd和[1, i]的gcd相同.
②gcd改变, 那么gcd至少变成gcd/2

我们相当于可以得到这样的序列 gcd([1, i₁]) = d₁, gcd([i₁ + 1, i₂]) = d₂, …, gcd[i_k + 1, n] = d_k+1
k的最大取值也只有log值域.

因此我们得出结论: 对于一段连续的区间求gcd的情况, 不同的gcd取值最多只有log值域个

---

这就很像HH的项链这道题. 同样是离线查询的手段来处理. 只不过这道题是对于每个位置, 我们要同时考虑k个值.

我们用同样的方式来分析一下, 当我们从左向右开始往树中插值时, 对于当前index位置, 有两种情况: ①当前值之前没有出现过 ②当前值之前在pos位置出现过.

对于情况①: 我们同样直接统计当前情况贡献即可.
对于情况②: 我们统计完当前贡献后, 还是要减去之前位置的贡献.

(感觉都是废话) 看分析的话, 请点击上面那个博客去看吧. 思路都是一样的.

还有一点需要考虑的是, 对于这题的每一个val, 它都存在于一段区间中, 我们应当考虑是在哪个位置加入当前val产生的贡献.

我们应当在最靠右的位置加入val产生的贡献., 因为我们如果已经要计算val的贡献了, 表明当前的查询[l, r]的右端点r已经覆盖到了当前点的情况, 因此我们只需要考虑左端点l.
假设val产生贡献的区间为[L, R], 如果我们在L处加入了val产生的贡献, 那么对于**l ∈[L + 1, R]**这一区间, 我们都会少计算val的贡献, 因此我们应当尽可能靠右, 即: 在R处加入val产生的贡献.

由于上述性质, 所以对于index位置求gcd时, 如果有连续某段的gcd相同, 我们应当尽可能靠右去记录其出现的位置(即: 记录其右端点的位置).

考虑求法, 我们可以对于每一个位置开一个vector, 表示以当前位置为结束点的所有gcd情况(gcd的值, 以及右端点的位置), 相当于 vector<pair<int,int>>类型, 这样每次通过index-1位置的gcd情况, 可以log级别的求出index位置的gcd情况.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 1E5 + 10;
int gcd(int a, int b) { return b ? gcd(b, a % b) : a; }
int n, m;
int w[N], res[N];
struct operation { //离线查询
	int l, r, id;
	bool operator< (const operation& t) const { return r < t.r; }
}; vector<operation> area;


int t[N];
int lowbit(int x) { return x & -x; }
void add(int x, int c) { for (int i = x; i <= n; i += lowbit(i)) t[i] += c; }
int ask(int x) {
	int res = 0;
	for (int i = x; i; i -= lowbit(i)) res += t[i];
	return res;
}
int ask(int l, int r) { return ask(r) - ask(l - 1); }

vector<pair<int, int>> v[N];
int main()
{
	while (~scanf("%d %d", &n, &m)) {
		rep(i, n) {
			t[i] = 0; v[i].clear(); //初始化
			scanf("%d", &w[i]);

			v[i].push_back({ w[i], i }); //记录初始合法情况
            int last = w[i]; //表示上一次的gcd结果
			for (auto& op : v[i - 1]) {
				int val = op.first, index = op.second;
				int temp = gcd(w[i], val);
				if (last != temp) v[i].push_back({ temp, index }); //和上次gcd结果不同
				last = temp; //更新gcd结果
			}
		}


		area.clear();
		rep(i, m) {
			int l, r; scanf("%d %d", &l, &r);
			area.push_back({ l, r, i });
		}
		sort(area.begin(), area.end());


		unordered_map<int, int> mp; //记录上次val这个值出现的位置
		int pos = 0;
		for (auto& op : area) {
			int l = op.l, r = op.r, id = op.id;
			while (pos + 1 <= r) {
				pos++;
				for (auto& op : v[pos]) {
					int val = op.first, index = op.second;
					add(op.second, 1);
					if (mp.count(val)) add(mp[val], -1);
					mp[val] = index;
				}
			}
			res[id] = ask(l, r);
		}

		rep(i, m) printf("%d\n", res[i]);
	}
	return 0;
}

其实我们这里也可以采用主席树的思路, 但是空间上需要nlog²n, 本题是无法接受的

END