Hackerrank (70)Range Modular Query 莫队+暴力

最新推荐文章于 2023-10-02 01:48:32 发布

Masamiiiii

最新推荐文章于 2023-10-02 01:48:32 发布

阅读量446

点赞数

分类专栏： Mo's Algorithm

本文链接：https://blog.csdn.net/Jeremy1149/article/details/68066482

版权

Mo's Algorithm 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

点击打开链接

题意:给出n个数,q次询问,q:给出l,r,x,y问[l,r]内ai=kx+y的个数.n,q,x,y,ai<=4e4;
n次询问,根据x,更新出ai,ai=kx+y的复杂度为nlogn
由于ai要在[l,r]内,离线后,利用莫队更新[l,r]内ai出现的次数,总的时间复杂度为O(n*sqrt(n)*logn))

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N=2e5+20;
const int M=4e4;
int n,Q,a[N];
int l,r,x,y;
int pos[N];
struct node{
	int l,r,x,y,id;
	int ans;
}q[N];
int mp[N];
bool cmp(node a,node b)
{
	if(pos[a.l]==pos[b.l])
		return a.r<b.r;
	return a.l<b.l;
}
bool cmp_id(node a,node b)
{
	return a.id<b.id;
}
void init()
{
	memset(mp,0,sizeof(mp));
	int block=(int)sqrt(n);
	for(int i=1;i<=n;i++)
		pos[i]=(i-1)/block+1;
}
void update(int p,int add)
{
	mp[a[p]]+=add;
}
void solve()
{
	for(int i=0,l=1,r=0;i<Q;i++)
	{
		for(;r<q[i].r;r++)
			update(r+1,1);
		for(;r>q[i].r;r--)
			update(r,-1);
		for(;l<q[i].l;l++)
			update(l,-1);
		for(;l>q[i].l;l--)
			update(l-1,1); 
		int ans=0;
		for(int k=0;k<=M;k+=q[i].x)
			ans+=mp[k+q[i].y];
		q[i].ans=ans;
	}
}
int main()
{
	while(cin>>n>>Q)
	{
		for(int i=1;i<=n;i++)
			scanf("%d",&a[i]);		
		init();
		for(int i=0;i<Q;i++)
		{
			scanf("%d%d%d%d",&q[i].l,&q[i].r,&q[i].x,&q[i].y);
			q[i].l++,q[i].r++;
			q[i].id=i;
		}
		sort(q,q+Q,cmp);
		solve();
		sort(q,q+Q,cmp_id);
		for(int i=0;i<Q;i++)
			printf("%d\n",q[i].ans);
	}
	return 0;
}