HDU - 6287 口算训练

problem

小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

思路

考虑到数据范围为1e5 质数大约有1e4个

判断一个区间的乘积是否为一个数d的倍数

判断方法是对d进行质因子分解，现在我们只关注这些素数

对于一个素数，看对于询问区间$[L,R]$中能整除这个素数几次（也就是区间内每个数质因子分解，若有这个素数则就是对应的指数（相加)，若没有就为0

现在的问题是怎么维护m次询问的结果

你会发现1e5个数就$1e5log(1e5)$个素数，只有这些值对答案有贡献，所以理论上是可以存下的。

而且只用关注1e4个素数

于是我们用二维vector<int> vec[1e4]

对于每一维，存的是有该质因子的每个数（共1e5个)的权值 即对应指数 并维护前缀和vector pre[1e4]

所以对于每个n(1e5)，在询问前预处理上述两个数组

时间$nlogn$

对于一个询问l,r,d

对d质因子分解后（可以提前预处理，也可以$）O(\sqrt n）$直接做

然后对于每个质数，找到对应的vec[prime] 二分l,r 直接用前缀和得到指数

看是否能整除即可

若所有d的质因子都能整除

则yes 否则no

时间复杂度$T*mlognlogn$

代码示例

注意二分细节 upper_bound 后-1 不然非法访问未知内存

以及维护pre 先 push占位数据0 方便pre

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

int debug_num=0;
#define debug cout<<"debug "<<++debug_num<<" :"

const int maxn=1e5+10;
int no[maxn];

int n,m;

vector<int> ans[maxn];//质因子
vector<int> bns[maxn];//对应指数
vector<int> vec[maxn];
vector<int> pre[maxn];

//先预处理每个数有哪些质因数
void init1(int n)//处理1000005个数本机跑了1.303s 评测机0.4s左右
{
    for(int i=2;i<=n;++i){
        if(ans[i].size()==0){
            for(int j=i;j<=n;j+=i){
                ans[j].push_back(i);
            }
        }
    }
}

void init2(int n)
{
    for(int i=2;i<=n;++i){
        int tp=i;
        for(int j=0;j<ans[i].size();++j){
            bns[i].push_back(0);
            while(tp%ans[i][j]==0)
            {
                bns[i][j]++;
                tp/=ans[i][j];
            }
        }
    }
}

void init()
{
    for(int i=0;i<=maxn;++i){
        vec[i].clear();
        vec[i].push_back(0);//占位0坐标
    }
    for(int i=0;i<=maxn;++i){
        pre[i].clear();
        pre[i].push_back(0);//前缀和占位
    }
    for(int i=1;i<=n;++i){
        int tp=no[i];
        for(int j=0;j<ans[tp].size();++j){
            int tmp=ans[tp][j];
            vec[tmp].push_back(i);
            int id=pre[tmp].size()-1;
            pre[tmp].push_back(pre[tmp][id]+bns[tp][j]);
        }
    }
}

bool check(int l,int r,int d,int id)//有了d和id就知道当前质数和指数
{
    int zhi=ans[d][id];
    int need=bns[d][id];
    //第一个大于等于l的id
    int id_l=lower_bound(vec[zhi].begin(),vec[zhi].end(),l)-vec[zhi].begin();
    int id_r=upper_bound(vec[zhi].begin(),vec[zhi].end(),r)-vec[zhi].begin()-1;
    int now=pre[zhi][id_r]-pre[zhi][id_l-1];
    if(now<need) return false;
    return true;
}

int main()
{
    //freopen("in.txt","r",stdin);
    ios::sync_with_stdio(false);
    init1(1e5);
    init2(1e5);
    int t;
    cin>>t;
    while(t--)
    {
        int l,r,d;
        cin>>n>>m;
        for(int i=1;i<=n;++i){
            cin>>no[i];
        }
        init();
//        for(int i=2;i<=30;++i){
//            cout<<"i="<<i<<" size: "<<vec[i].size()<<endl;
//            if(vec[i].size()){
//                for(int j=0;j<vec[i].size();++j){
//                    cout<<vec[i][j]<<" ";
//                }
//                cout<<endl;
//            }
//        }
        for(int i=1;i<=m;++i){
            cin>>l>>r>>d;
            int flag=1;
            for(int j=0;j<ans[d].size();++j){
                if(!check(l,r,d,j)){
                    flag=0;
                    break;
                }
            }
            if(flag) cout<<"Yes"<<endl;
            else cout<<"No"<<endl;
        }
    }
    return 0;
}