C. Complex Market Analysis

最新推荐文章于 2024-06-01 18:05:44 发布

jikelk

最新推荐文章于 2024-06-01 18:05:44 发布

阅读量77

点赞数

分类专栏： c++ 文章标签：算法数据结构

本文链接：https://blog.csdn.net/jikelk/article/details/128498852

版权

c++ 专栏收录该内容

83 篇文章 0 订阅

订阅专栏

传送门

题意：

给你一个数组和一个e，问有多少对 $(i,k)$ 能够满足其为质数

$(i,k)$ 为 $s[i]*s[i+e]*s[i+2e]*...*s[i+ke]$ 的值

思路：

因为要求某个子区间为质数，那么这其中一定有且只有一个质数。可以提前用筛法预处理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;
}