1561 : [算法竞赛进阶指南]Interval GCD

1561 : [算法竞赛进阶指南]Interval GCD

Time Limit:2 Sec Memory Limit:128 MB
Problem Tags:线段树 Manager:zhbink

Submit C++ Solution Edit

Description
给定一个长度为N的数列A，以及M条指令，每条指令可能是以下两种之一：

1、“C l r d”，表示把 A[l],A[l+1],…,A[r] 都加上 d。

2、“Q l r”，表示询问 A[l],A[l+1],…,A[r] 的最大公约数(GCD)。

对于每个询问，输出一个整数表示答案。

Input
第一行两个整数N,M。

第二行N个整数A[i]。

接下来M行表示M条指令，每条指令的格式如题目描述所示。

数据范围：
N≤500000,M≤100000

Output
对于每个询问，输出一个整数表示答案。

每个答案占一行。

Sample Input
5 5
1 3 5 7 9
Q 1 5
C 1 5 1
Q 1 5
C 3 3 6
Q 2 4
Sample Output
1
2
4

下面这个是自己敲的直接是按照线段树来敲的超时了，其实思路也有误，
C操作中的加数d我都按照正数来计算了（其实是所有的数都按照正数来计算了）
下面有正确的程序

#include<iostream>
using namespace std;
const int maxn=500005;
using ll=long long;
struct  Segment{
	int gcd;
	int l,r;
	Segment(){
		gcd=l=r=0;
	}
}t[maxn<<2];
int n,m;
int a[maxn];
int GCD(int a,int b)
{   
return b ? GCD(b, a % b) : a;
//if(x<=0)return y;
//if(y<=0)return x;
//	int maxx,minn,i=0;
//	maxx=max(x,y);
//	minn=min(x,y);
//	if(maxx%minn==0)return minn;
//   for( i=minn/2+1;i>=1;i--)
//   if(maxx%i==0&&minn%i==0)return i;
   
}
void bulid(int p,int l,int r){//建树正确 
t[p].l=l;
t[p].r=r;
if(l==r)
{
	t[p].gcd=a[l];
	return;
	}	
	int mid=(l+r)>>1;
	bulid(p<<1,l,mid);//建立左子树 
	bulid(p<<1|1,mid+1,r);//建立右子树 
	t[p].gcd=GCD(t[p<<1].gcd,t[p<<1|1].gcd);

}
Segment ask(int p,int l,int r)
{
	if(l<=t[p].l&&r>=t[p].r)
	{
		return t[p];
	}
	Segment a,b,ret;
	int mid=(t[p].l+t[p].r)>>1;
	if(l<=mid)a=ask(p<<1,l,r);//向左子树查询
	if(r>mid)b=ask(p<<1|1,l,r);//向右子树查询
	ret.gcd=GCD(a.gcd,b.gcd);
	return ret; 	
}
void change(int p,int l,int r,int v)
{
	if(t[p].l==t[p].r)
	{
		t[p].gcd+=v;
		return;
	}
	int mid=(t[p].l+t[p].r)>>1;
	if(l<=mid)
	{
		change(p<<1,l,r,v);	
	}
	if(r>mid)
	{
		change(p<<1|1,l,r,v);	
	}
	t[p].gcd=GCD(t[p<<1].gcd,t[p<<1|1].gcd);
}
int main(){
	freopen("qwe.txt","r",stdin);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	bulid(1,1,n);	
	while(m--){
		char c[2];
		scanf("%s",c);
		if(c[0]=='Q'){
			int k1,k2;
			scanf("%d %d",&k1,&k2);	
			printf("%d\n",ask(1,k1,k2).gcd);
		}
		if(c[0]=='C')
		{    int k1,k2,k3;
			scanf("%d %d %d",&k1,&k2,&k3);
			change(1,k1,k2,k3);
		}	
	}
	return 0;
}
 

以下是千方百计得到的标准代码

//待我弄懂之后会写注释的

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
using ll = long long;
const int maxn = 5e5 + 10;
struct SegmentTree
{
	int l, r;
	ll dat; // 差分序列的区间gcd
} tree[maxn << 2];
ll a[maxn], d[maxn], c[maxn];
int n, m;
 
ll gcd(ll a, ll b)
{
	return b ? gcd(b, a % b) : a;
}
 
void pushup(int p)
{
	tree[p].dat = gcd(tree[p << 1].dat, tree[p << 1 | 1].dat);
}
 
void build(int p, int l, int r)
{
	tree[p].l = l, tree[p].r = r;
	if (l == r)
	{
		tree[p].dat = d[l];
		return;
	}
	int mid = (l + r) >> 1;
	build(p << 1, l, mid);
	build(p << 1 | 1, mid + 1, r);
	pushup(p);
}
 
void change(int p, int x, ll v)
{
	if (tree[p].l == tree[p].r)
	{
		tree[p].dat += v;
		return;
	}
	int mid = (tree[p].l + tree[p].r) >> 1;
	if (x <= mid)
		change(p << 1, x, v);
	else
		change(p << 1 | 1, x, v);
	pushup(p);
}
 
ll ask(int p, int l, int r)
{
	if (l <= tree[p].l && r >= tree[p].r)
		return abs(tree[p].dat);
	int mid = (tree[p].l + tree[p].r) >> 1;
	ll val = 0;
	if (l <= mid)
		val = gcd(val, ask(p << 1, l, r));
	if (r > mid)
		val = gcd(val, ask(p << 1 | 1, l, r));
	return abs(val);
}
 
ll sum(int x)
{
	ll y = 0;
	for (; x; x -= x & -x)
		y += c[x];
	return y;
}
 
void add(int x, ll y)
{
	for (; x <= n; x += x & -x)
		c[x] += y;
}
 
int main()
{
	cin >> n >> m;
	for (int i = 1; i <= n; i++)
	{
		scanf("%lld", &a[i]);
		d[i] = a[i] - a[i - 1];
	}
	build(1, 1, n);
	for (int i = 1; i <= m; i++)
	{
		char str[2];
		scanf("%s", str);
		int l, r;
		scanf("%d%d", &l, &r);
		if (str[0] == 'Q')
		{
			ll al = a[l] + sum(l);
			ll val = l < r ? ask(1, l + 1, r) : 0;
			printf("%lld\n", gcd(al, val));
		}
		else
		{
			ll delta;
			scanf("%lld", &delta);
			change(1, l, delta);
			if (r < n)
				change(1, r + 1, -delta);
			add(l, delta);
			add(r + 1, -delta);
		}
	}
}