BZOJ 2506: calc 权值分块

2506: calc

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 573  Solved: 304
[Submit][Status][Discuss]

Description 

         给一个长度为n的非负整数序列A1,A2,…,An。现有m个询问，每次询问给出l,r,p,k，问满足l<=i<=r且Ai mod p = k的值i的个数。

Input

         第一行两个正整数n和m。

         第二行n个数，表示A1,A2,…,An。

         以下m行，每行四个数分别表示l,r,p,k。满足1<=l<=r<=n。

Output

         对于每个询问，输出一行，表示可行值i的个数。

Sample Input

5 2
1 5 2 3 7
1 3 2 1
2 5 3 0

Sample Output

2
1

HINT

数据范围：
         0<n,m<=10^5，任意1<=i<=n满足Ai<=10^4，0<p<=10^4，0<=k<p。

思想江化了啊

竟然没想到。。。太弱了。。。哇哇哇

肯定先离个线

p<=100随便做

p>=100时%p==k的数最多就100个 开个桶就行了。。。

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100;

struct Query{int opt,pos,p,k,id;friend bool operator <(const Query &x,const Query &y){return x.pos<y.pos;}}q[N<<1];

int f[110][110],num[N];

inline void insert(int x)
{
	for(int i=1;i<=100;++i)f[i][x%i]++;
		num[x]++;
}

int a[N],ans[N];

int main()
{
	int n=read(),Q=read();
	register int i,j,l,r,p,k;
	for(i=1;i<=n;++i)a[i]=read();
	int tot=0;
	for(i=1;i<=Q;++i)
	{
		l=read();r=read();p=read();k=read();
		q[++tot].pos=l-1;q[tot].opt=-1;q[tot].p=p;q[tot].k=k;q[tot].id=i;
		q[++tot].pos=r;q[tot].opt=1;q[tot].p=p;q[tot].k=k;q[tot].id=i;
	}
	sort(q+1,q+1+tot);
	int now=1;
	for(i=1;i<=tot;++i)
	{
		while(now<=q[i].pos)insert(a[now]),now++;
		if(q[i].p<=100)
			ans[q[i].id]+=q[i].opt*f[q[i].p][q[i].k];
		else 
			for(j=q[i].k;j<=10000;j+=q[i].p)ans[q[i].id]+=q[i].opt*num[j];
	}
	for(i=1;i<=Q;++i)print(ans[i]),puts("");
	return 0;
}