ACwing 区间最大公约数题解 线段树(附证明)
https://www.acwing.com/problem/content/247/
题目:
给定一个长度为 N 的数列 A,以及 M 条指令,每条指令可能是以下两种之一:
-
C l r d
,表示把 A[l],A[l+1],…,A[r] 都加上 d。 -
Q l r
,表示询问 A[l],A[l+1],…,A[r] 的最大公约数(GCD)。
对于每个询问,输出一个整数表示答案。
输入格式:
第一行两个整数 N,M。
第二行 N 个整数 A[i]。
接下来 M 行表示 M 条指令,每条指令的格式如题目描述所示。
输出格式:
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围:
N≤500000,M≤100000, 1≤A[i]≤10^{18}, |d|≤10^{18}
样例:
data.1
5 5
1 3 5 7 9
Q 1 5
C 1 5 1
Q 1 5
C 3 3 6
Q 2 4
1
2
4
data.2
5 5
1 3 5 7 9
Q 1 5
C 1 5 1
Q 5 5
C 3 3 6
Q 2 4
1
10
4
思路:
首先区间修改,考虑线段树,分析如何处理怎么维护区间的 gcd,容易想到: gcd(a,b,c)=gcd(gcd(a,b),gcd(b,c)) ,但是由于存在区间修改,我们需要处理如何从 gcd(a,b,c) 推导出 gcd(a+1,b+1,c+1)?结论是利用差分,因为有 gcd(a,b)=gcd(a,b-a),所以有:
证明:
证明 gcd(a,b) = gcd(a,b-a):
令gcd(a,b)=g,所以 \left\{\begin{matrix} a=k_1g \\ b=k_2g \end{matrix}\right. ,且 k_1 与 k_2 互质,所以 b=(k_2-k_1)g,若证 gcd(a,b)=gcd(a,b-a),即证:gcd(a,b-a)=g=gcd(a,b),即证:k_1 与 (k_2-k_1) 互质。
假设 k_1 与 (k_2-k_1) 不互质,存在最大公因数 g',g'≠1,则有:\left\{\begin{matrix} k_1=xg' \\ k_2-k_1=yg' \end{matrix}\right. 。
即 (k_2-k_1)g'+k_1g'=(x+y)g'=k_2,那么有 gcd(k_1,k_2)=g', g'≠1,则 k_1⊥(k_2-k_1), 那么有 gcd(a,b-a)=g=gcd(a,b) 成立。
证明 gcd(a,b,c)=gcd(a,b-a,c-b)
用 (a,b) 表示 gcd(a,b)。
线段树 tr 对差分进行维护,我们想要修改原数组中 A[l,r],都加上d,就相当于在差分数组上 b[l]+=d
,b[r+1]-=d
,对应线段树就是 modify(1, l, d)
,以及 modify(1, r+1, -d)
,需要注意比较 r+1 和线段树边界 N 的大小,若 r+1>N ,则直接不需要操作。 由于 gcd(a,b,c)=gcd(a,b-a,c-b),所以我们需要知道第一个的真实值,即对差分数组求前缀和,对应线段树操作就是 askSum(1,1,l)
,并且和 [l+1,r] 求最大公约数 gcd,即 askGcd(1,l+1,r)
,因为差分操作使得 gcd 存在负数,则是不符合的,所以输出需要取绝对值。
由数据范围知需要开 long long
。
pushup
的实现:
void pushup(seg &u, seg &l, seg &r) {
u.sum = l.sum+r.sum;
u.gcd = __gcd(l.gcd, r.gcd);
}
Coding 时间到
#include <iostream> #include <algorithm> #define int long long using namespace std; constexpr int N = 5e5 + 13; struct seg{ int l, r; int sum, gcd; }tr[N*4]; int n, a[N], b[N]; void pushup(seg &u, seg &l, seg &r) { u.sum = l.sum+r.sum; u.gcd = __gcd(l.gcd, r.gcd); } void pushup(int u) { pushup(tr[u], tr[u<<1], tr[u<<1|1]); } void build(int u, int l, int r) { tr[u].l=l, tr[u].r=r; if (l==r) tr[u].sum=tr[u].gcd=b[l]; else { int mid=l+r>>1; build(u<<1, l, mid), build(u<<1|1, mid+1, r); pushup(u); } } void modify(int u, int pos, int x) { if (pos == tr[u].l && tr[u].r == pos) { tr[u].sum += x, tr[u].gcd += x; } else { int mid = tr[u].l+tr[u].r>>1; if (pos <= mid) modify(u<<1, pos, x); else modify(u<<1|1, pos, x); pushup(u); } } int askSum(int u, int l, int r) { if (l <= tr[u].l && tr[u].r <= r) { return tr[u].sum; } int mid = tr[u].l+tr[u].r>>1; int ans = 0; if (l <= mid) ans+=askSum(u<<1, l, r); if (r > mid) ans+=askSum(u<<1|1, l, r); return ans; } int askGcd(int u, int l, int r) { if (l <= tr[u].l && tr[u].r <= r) { return tr[u].gcd; } int mid = tr[u].l + tr[u].r >> 1; if (r <= mid) return askGcd(u<<1, l, r); else if (l > mid) return askGcd(u<<1|1, l, r); else return __gcd(askGcd(u<<1,l,r), askGcd(u<<1|1,l,r)); } int solve() { int q; cin >> n >> q; for (int i=1;i <= n;i ++ ) { cin>>a[i]; b[i]=a[i]-a[i-1]; } build(1, 1, N-3); // for (int i=1;i <= n;i ++ ) // cout<<askSum(1, 1, i)<<" \n"[i==n]; for (int i=1;i <= q;i ++ ) { string op; cin >> op; if (op[0] == 'C') { int l, r, d; cin >> l >> r >> d; modify(1, l, d); if (r+1 < N) modify(1, r+1, -d); } else { int l, r; cin >> l >> r; int step = askSum(1, 1, l); int g = (r-l)?__gcd(step, askGcd(1, l+1, r)):step; cout << abs(g) << '\n'; } // cout<<"i:? "<<'\n'; // for (int i=1;i <= n;i ++ ) // cout<<askSum(1,1,i)<<" \n"[i==n]; } return 1; } signed main() { cin.tie(0) -> sync_with_stdio(0); int T(1); //cin>>T; while (T -- ) solve(); return 0; }