C. Complex Market Analysis

传送门

题意：

给你一个数组和一个e，问有多少对能够满足其为质数

为的值

思路：

因为要求某个子区间为质数，那么这其中一定有且只有一个质数。可以提前用筛法预处理1e6 的质数，然后从每一个质数开始往左或者往右不断移动e次，直到当前元素不为1为止，那么结果就是前面的1数量加后面1的数量加两数相乘。

#include<cstdio>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <string>
#include <math.h>
#include<vector>
#include<queue>
#include<map>
#define sc_int(x) scanf("%d", &x)
#define sc_ll(x) scanf("%lld", &x)
#define pr_ll(x) printf("%lld", x)
#define pr_ll_n(x) printf("%lld\n", x)
#define pr_int_n(x) printf("%d\n", x)
#define ll long long 
using namespace std;

const int N=3000000+100;
int n ,m,h;
ll s[N],p[N];
bool st[N];int tim;

void solve()
{
	int e;
	cin>>n>>e;
	for(int i =1;i<=n;i++)cin>>s[i];
	ll res=0;
	for(int i =1;i<=n;i++)
	{
		if(!st[s[i]])
		{
			ll a=0,b=0;
			for(int j =i+e;j<=n;j+=e)
			{
				if(s[j]==1)
				{
					res++;
					a++;
				}
				else break;
			}

			for(int j=i-e;j>=1;j-=e)
			{
				if(s[j]==1)
				{
					res++;
					b++;
				}
				else break;
			}
			res+=a*b;
		}
	}
	cout<<res<<endl;
	
	return ;
}


void init()
{
	st[1]=1;
	for(int i =2;i<=2e6;i++)
	{
		if(!st[i])p[++tim]=i;
		for(int j =1;p[j]<=2e6/i;j++)
		{
			st[i*p[j]]=1;
			if(i%p[j]==0)break;
		}
	}
}

int main()
{
	int t;
	init();
	sc_int(t);
	while(t--)solve();
	
	return 0;
}