CCPC2018女生赛口算训练6287

#include <cstdio>
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
const int N = 1e5 + 5;
vector<int> prime[N];//二维的质因数容器，每个prime[N]存的是输入序列中的下标

//二分法查找质因数，用lower_bound和upper_bound函数
int query(int i, int l, int r) {
	return upper_bound(prime[i].begin(), prime[i].end(), r) - lower_bound(prime[i].begin(), prime[i].end(), l);
	//在目标序列段：[右边的最后一个数]下标-[左边第一个数]下标
}
/*
1、查找目标数的质因子,先cnt=0,再while，cnt++累加相同质因子的个数
2、每找完一个因子数，就跟prime[N]中的指定序列的因子个数相比，小于则false
3、如果目标数本身是质数、或者数太小进不去for循环、或者cnt一直=0，都会错过上面的false条件，所以要用新的if判断，
当目标数不为1且目标数本身在输入序列质因子中不存在，返回false
*/
bool half_find(int l, int r, int d) {
	for (int i = 2; i*i <= d; i++) {
		int cnt = 0;//每次循环先置为0
		while (d%i == 0) {
			d /= i;//目标数变小
			cnt++;
		}
		if (cnt > query(i, l, r)) return false;//每一个数判断一次，目标数的该因子是否大于序列数的该因子
	}
	if ((d != 1) && (query(d, l, r) < 1)) return false;//三种情况：数太小进不去for循环、d为质数、排除了“任何数都是1的倍数”的情况
	return true;
}

int main() {
	int T, n, m, a[N];
	int l, r, d;
	scanf("%d", &T);
	while (T--) {
		scanf("%d %d", &n, &m);

		for (int i = 1; i < N; i++) {
			prime[i].clear();//清空vector!!!
		}

		for (int i = 1; i <= n; i++) {//题目是从1开始
			scanf("%d", &a[i]);
			for (int j = 2; j*j <= a[i]; j++) {//注意是从2开始的
				while (a[i] % j == 0) {
					a[i] /= j;//序列数变小
					prime[j].push_back(i);//在prime[因子]中放入序列的下标（for循环会导致下标按顺序存储）
				}
			}
			if (a[i] > 1) prime[a[i]].push_back(i);//如果目标数是质数、或者目标数进不去for循环的情况,注意：存的是自己不是j
		}

		//输入问题
		for (int k = 1; k <= m; k++) {
			scanf("%d %d %d", &l, &r, &d);
			if (half_find(l, r, d))    printf("Yes\n");
			else printf("No\n");
		}
	}
	return 0;
}

Problem - 6287 (dingbacode.com)