hdu 5381 The sum of gcd 莫队 + DP

第一次用莫队A题。

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=5381

#include <map>
#include <set>
#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;

const int MAXN = 10005;

struct Node {
	int x,y;
	int id;
}t[MAXN];

struct Seg{
	int Lseg[40],Rseg[40],value[40];
	int len;
}s1[MAXN],s2[MAXN];


int a[MAXN];
int unit;
long long ans[MAXN];
int n ,m;

bool cmp(Node a1,Node a2) {
	if(a1.x/unit == a2.x/unit) return a1.y < a2.y;
	return a1.x/unit < a2.x/unit;
}

int gcd(int a1,int a2) {
	while(a2) {
		int temp = a1 % a2;
		a1 = a2;
		a2 = temp;
	}
	return a1;
}

long long calc_y(int x,int y) {
	long long temp = 0;
	//以y+1为终点。
	for(int k = s1[y+1].len ; k >= 1 ; k--) {
		if(s1[y+1].Lseg[k] >= x) {
			temp += (long long )(s1[y+1].Rseg[k] - s1[y+1].Lseg[k] + 1) * (long long )s1[y+1].value[k];
		}
		else {
			temp += (long long )(s1[y+1].Rseg[k] - x + 1) * (long long )s1[y+1].value[k];
			break;
		}
	}
	//printf("calc_y : %d\n",temp);
	return temp; 
}

long long calc_x(int x,int y) {
	long long temp = 0;
	//以x-1为起点。
	for(int k = s2[x-1].len ; k >= 1; k--) {
		if(s2[x-1].Rseg[k] <= y) {
			temp += (long long)(s2[x-1].Rseg[k] - s2[x-1].Lseg[k] + 1) * (long long)s2[x-1].value[k];
		}
		else {
			temp += (long long)(y - s2[x-1].Lseg[k] + 1) * (long long)s2[x-1].value[k];
			break;
		}
	}
	//printf("x : %d y : %d temp : %d\n",x,y,temp);
	//printf("calc_x : %d\n",temp);
	return temp; 
}

void Solve() {
	int x= 1;
	int y = 0;
	long long temp = 0;
	for(int i = 1 ; i <= m ; i++) {
		while(y < t[i].y) {
			y++;
			temp += calc_y(x,y-1);
		}
		while(y > t[i].y) {
			temp -= calc_y(x,y-1);
			y--;
		}
		while(x < t[i].x) {
			temp -= calc_x(x+1,y);
			x++;
		}
		while(x > t[i].x) {
			x --;
			temp += calc_x(x+1,y);
		}
		//printf("temp : %lld\n",temp);
		ans[t[i].id] = temp;
	}
}

void Deal_with() {
	int T;
	scanf("%d",&T);
	while(T--) {
		memset(s1,0,sizeof(s1));
		memset(s2,0,sizeof(s2));
		scanf("%d",&n);
		for(int i = 1 ; i <= n ; i++) {
			scanf("%d",a+i);
		}
		//以i为终点的GCD区间。
		for(int i = 1 ; i <= n ; i++) {
			int k = 0 ; //区间的下标。
			int temp = -1;//前一个区间和a[i]的GCD值。
			int temp2;//当前区间和a[i]的GCD值。
			for(int j = 1 ; j <= s1[i-1].len ; j++) { //枚举前一个区间。
				temp2 = gcd(a[i],s1[i-1].value[j]);
				if(temp2 == temp) { //前一个区间的GCD和现在区间的GCD相等，就合并。
					s1[i].Rseg[k] = s1[i-1].Rseg[j]; 
				}
				else { // 前一个区间的GCD和现在区间的GCD不相等，开始一个新的区间。
					k ++;
					s1[i].Lseg[k] = s1[i-1].Lseg[j];
					s1[i].Rseg[k] = s1[i-1].Rseg[j];
					s1[i].value[k] = temp2;
					s1[i].len ++;
				}
				temp = temp2;
			}
			//处理完了区间之后，还要处理端点。
			temp2 = a[i];
			if(temp2 == temp) { //前一个区间的GCD和现在区间的GCD相等，就合并。
				s1[i].Rseg[k] = i; 
			}
			else { // 前一个区间的GCD和现在区间的GCD不相等，开始一个新的区间。
				k ++;
				s1[i].Lseg[k] = i;
				s1[i].Rseg[k] = i;
				s1[i].value[k] = temp2;
				s1[i].len ++;
			}
			//printf("s1 len : %d\n",s1[i].len);
			//for(int j = 1 ; j <= s1[i].len ; j ++) {
			//	printf("Left : %d Right : %d\n",s1[i].Lseg[j],s1[i].Rseg[j]);
			//}
		}

		//以i为起点的GCD区间。
		for(int i = n ; i >= 1 ; i--) {
			int k = 0 ; //区间的下标。
			int temp = -1;//前一个区间和a[i]的GCD值。
			int temp2;//当前区间和a[i]的GCD值。
			for(int j = 1 ; j <= s2[i+1].len ; j++) { //枚举前一个区间。
				temp2 = gcd(a[i],s2[i+1].value[j]);
				if(temp2 == temp) { //前一个区间的GCD和现在区间的GCD相等，就合并。
					s2[i].Lseg[k] = s2[i+1].Lseg[j]; 
				}
				else { // 前一个区间的GCD和现在区间的GCD不相等，开始一个新的区间。
					k ++;
					s2[i].Lseg[k] = s2[i+1].Lseg[j];
					s2[i].Rseg[k] = s2[i+1].Rseg[j];
					s2[i].value[k] = temp2;
					s2[i].len ++;
				}
				temp = temp2;
			}
			//处理完了区间之后，还要处理端点。
			temp2 = a[i];
			if(temp2 == temp) { //前一个区间的GCD和现在区间的GCD相等，就合并。
				s2[i].Lseg[k] = i; 
			}
			else { // 前一个区间的GCD和现在区间的GCD不相等，开始一个新的区间。
				k ++;
				s2[i].Lseg[k] = i;
				s2[i].Rseg[k] = i;
				s2[i].value[k] = temp2;
				s2[i].len ++;
			}
			//printf("a[i] : %d s2 : len : %d\n",a[i],s2[i].len);
			//for(int j = 1 ; j <= s2[i].len ; j ++) {
			//	printf("Left : %d Right : %d\n",s2[i].Lseg[j],s2[i].Rseg[j]);
			//}
		}

		scanf("%d",&m);
		for(int i = 1 ; i <= m ; i++) {
			scanf("%d %d",&t[i].x,&t[i].y);
			t[i].id = i;
		}
		unit = sqrt(n);
		sort(t+1,t+m+1,cmp);
		Solve();
		for(int i = 1 ; i <= m ; i++) {
			printf("%lld\n",ans[i]);
		}
	}
}

int main(void) {
    //freopen("1002.in","r",stdin);
    //freopen("a.out","w",stdout);
    Deal_with();
    return 0;
}