hdu6287 口算训练（唯一分解定理+二分）

口算训练

Time Limit: 8000/4000 MS (Java/Others)    Memory Limit: 512000/512000 K (Java/Others)
Total Submission(s): 2925    Accepted Submission(s): 648

 

Problem Description

小Q非常喜欢数学，但是他的口算能力非常弱。因此他找到了小T，给了小T一个长度为n的正整数序列a1,a2,...,an，要求小T抛出m个问题以训练他的口算能力。

每个问题给出三个正整数l,r,d，小Q需要通过口算快速判断al×al+1×...×ar−1×ar是不是d的倍数。

小Q迅速地回答了出来，但是小T并不知道正确答案是什么，请写一个程序帮助小T计算这些问题的正确答案。

 

 

Input

第一行包含一个正整数T(1≤T≤10)，表示测试数据的组数。

每组数据第一行包含两个正整数n,m(1≤n,m≤100000)，分别表示序列长度以及问题个数。

第二行包含n个正整数a1,a2,...,an(1≤ai≤100000)，表示序列中的每个数。

接下来m行，每行三个正整数l,r,d(1≤l≤r≤n,1≤d≤100000)，表示每个问题。

 

 

Output

对于每个问题输出一行，若是倍数，输出Yes，否则输出No。

 

 

Sample Input

1 5 4 6 4 7 2 5 1 2 24 1 3 18 2 5 17 3 5 35

 

 

Sample Output

Yes No No Yes

 

 

Source

"字节跳动杯"2018中国大学生程序设计竞赛-女生专场

 解题思路

唯一分解定理：任何一个大于1的整数n都可以分解成若干个素因数的连乘积，且这种分解是惟一的。如果a是b的倍数，那么对于b分解的素因数，在a中出现的次数都大于等于在b中出现的次数。

在输入T前先预处理，分解所有可能的an值，及1-100000，把素因数（可重复）存入对应的vector里。

然后开始处理输入数据。对于ai，在它的所有素因数后面存入i，那么此时，最后一个i的下标 + 1就是前i个数的素因数里出现的此素因数的个数。所以对于d，我们只要用二分查询d的所有素因数在l到r范围内出现的次数就可以了。

代码如下

#include <iostream>
#include <vector>
#include <cstdio>
#include <cmath>
#include <algorithm>
#define maxn 100005
using namespace std;
vector<int> vec[maxn];
vector<int> p[maxn];
vector<int> prime;
int main()
{
	//打印素数表 
    for(int i = 2; i <= 100000; i ++){
        bool flg = true;
        for(int j = 2; j <= sqrt(i); j ++){
            if(i % j == 0){
                flg = false;
                break;
            }
        }
        if(flg)
            prime.push_back(i);
    }
    //预处理得到所有分解 
    for(int i = 2; i <= 100000; i ++){
        int t = i;
        for(int j = 0; j < prime.size() && t != 1; j ++){
            while(t % prime[j] == 0){
                p[i].push_back(prime[j]);
                t /= prime[j];
            }
        }
    }
    int T;
    scanf("%d", &T);
    while(T --){
        int n, m;
        scanf("%d%d", &n, &m);
        for(int i = 1; i <= n; i ++){
            int x;
            scanf("%d", &x);
            for(int j = 0; j < p[x].size(); j ++){
                vec[p[x][j]].push_back(i);
            }
        }
        for(int i = 1; i <= m; i ++){
            int l, r, d;
            scanf("%d%d%d", &l, &r, &d);
            bool flg = true;
            for(int j = 0; j < p[d].size(); j ++){
                int cnt = 1;
                //得到此素因数在d中出现的次数 
                while(j + 1 < p[d].size() && p[d][j] == p[d][j + 1]){
                    j ++;
                    cnt ++;
                }
                //二分 
                int a = lower_bound(vec[p[d][j]].begin(), vec[p[d][j]].end(), l) - vec[p[d][j]].begin() - 1;
                int b = upper_bound(vec[p[d][j]].begin(), vec[p[d][j]].end(), r) - vec[p[d][j]].begin() - 1;
                //比较次数，若下于则说明不是d的倍数 
                if(b - a < cnt){
                    flg = false;
                    break;
                }
            }
            if(flg)
                printf("Yes\n");
            else
                printf("No\n");
        }
        for(int i = 0; i < maxn; i ++)
            vec[i].clear();
    }
    return 0;
}