链接:题目传送门
来源:牛客网
数列互质
时间限制:C/C++ 6秒,其他语言12秒
空间限制:C/C++ 524288K,其他语言1048576K
64bit IO Format: %lld
题目描述
给出一个长度为 n 的数列 { a[1] , a[2] , a[3] , … , a[n] },以及 m 组询问 ( l[i] , r[i]
, k[i])。
求数列下标区间在 [ l[i] , r[i] ] 中有多少数在该区间中的出现次数与 k[i] 互质(最大公约数为1)。
输入描述:
第一行,两个正整数 n , m (1 ≤ n, m ≤ 50000)。
第二行,n 个正整数 a[i] (1 ≤ a[i] ≤ n)描述这个数列。
接下来 m 行,每行三个正整数 l[i] , r[i] , k[i] (1 ≤ l[i] ≤ r[i] ≤ n, 1 ≤ k[i] ≤ n),描述一次询问。
输出描述:
输出 m 行,即每次询问的答案。
输入:
10 5
1 1 1 1 1 2 2 2 2 2
4 7 2
4 7 3
4 8 2
4 8 3
3 8 3
输出:
0
2
1
1
0
**思路:**用莫队找出区间内各数的个数,用set记录所有的数,遍历一下即可。
莫队入门题目:莫队入门–区间问题
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
const ll N=50010;
const ll M=100010;
const ll S=1000;
ll n,m,a[N],cnt[M],nr,nl,now,ans[M];
set<ll> s;
struct node
{
ll l,r,id,id1;
bool operator<(const node &rh)const
{
if((l-1)/S!=(rh.l-1)/S)
{
return (l-1)/S<(rh.l-1)/S;
}
return r<rh.r;
}//定义优先级
}query[N];
ll gcd(ll a,ll b)
{
return b ? gcd(b,a%b) : a;
}
void add(ll x,ll y)
{
++cnt[x];
}
void del(ll x,ll y)
{
--cnt[x];
}
void move_to(ll tl,ll tr,ll tt)
{
while(nr<tr)add(a[++nr],tt);
while(nl>tl)add(a[--nl],tt);
while(nr>tr)del(a[nr--],tt);
while(nl<tl)del(a[nl++],tt);
for(set<ll>::iterator it=s.begin();it!=s.end();it++)
{
if(gcd(cnt[*it],tt)==1&&cnt[*it]!=0)
{
now++;
}
}//记录结果
}//区间维护
int main()
{
cin>>n>>m;
for(ll i=1;i<=n;i++)
{
cin>>a[i];
s.insert(a[i]);
}
for(ll i=1;i<=m;i++)
{
scanf("%lld%lld%lld",&query[i].l,&query[i].r,&query[i].id1);
query[i].id = i;
}
sort(query+1,query+m+1);
for(ll i=1;i<=m;i++)
{
move_to(query[i].l,query[i].r,query[i].id1);
ans[query[i].id]=now;
now=0;
}
for(ll i=1;i<m;i++)
{
cout<<ans[i]<<endl;
}
cout<<ans[m];//输出结果
return 0;
}