PAT2021春季 乙级 素数等差数列 有几个pintia

2021 春季 乙级

最大公差的素数等差数列

我设了两个数组，一个只存0和1，判断它是不是素数；一个依次列出所有不超过上确界的素数
这个地方鸭，如果公差gap从1开始枚举是会超时的，从首项开始依次和后面的素数做差，这样得到gap，才不会超时啊

#include<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;

class node
{
public:
	int gap,a;
	node(int g,int a):gap(g),a(a){
	}
};
bool cmp(node& n1,node& n2)
{
	if(n1.gap==n2.gap) return n1.a>n2.a;
	return n1.gap>n2.gap;
}
bool is_prime(int x)
{
	if(x<2) return false;
	for(int i=2;i<=sqrt(x);i++)
	{
		if(x%i==0) return false;
	}
	return true;
}

int main()
{
	int n,maxp;
	scanf("%d%d",&n,&maxp);
	int table[100005]={0},primes[100005]={0},k=0;
	for(int i=2;i<=maxp;i++)
	{
		if(is_prime(i))
		{
			table[i]=1;
			primes[k++]=i;
		}
	}
	
	if(n==1) 
    {
        printf("%d",primes[k-1]);
        return 0;
    }
	
	int gap,kk=2;
	bool flag;
	vector<node>ans;
	for(int i=0;i<k-1;i++)
	{// 2 3 5 7 11 13
        kk=i+2;
		for(gap=1;kk<k-1;gap=primes[kk++]-primes[i])
		{
         // cout<<gap<<endl;
			flag=true;
			for(int j=0;j<n;j++)
			{
				if(table[primes[i]+j*gap]!=1)
				{
					flag=false;
					break;
				}
			}
			
			if(!flag) continue;
			else ans.push_back(node(gap,primes[i]));
            
		}
		
	}
	sort(ans.begin(),ans.end(),cmp);
	if(ans.size()==0)  printf("%d",primes[k-1]);
	else
	{
		//cout<<ans[0].a<<" "<<ans[0].gap<<endl; 
		for(int i=0;i<n;i++)
		{
			if(i) printf(" ");
			printf("%d",ans[0].a+i*ans[0].gap);
		}
	}
}

一句话中有几个pintia/Pintia

输入不一定符合规范qaq 比如这样：pintia,pintia,pintia 即是说，逗号或者句号后可能无空隔，也可能有空隔（虽然正常来说英语的逗号后面要加空格的）
可以先把句子存下来，然后一个字符一个字符地遍历，一旦找到分隔符（空格，句号或者逗号），就把分隔符前的字符串扒拉下来扔到栈里，全部处理完毕后，把栈里的元素拿出来检查一遍，就行啦qaq
考试的时候没想到它逗号后可能不会有空格，然后，，，写得就很乱啊TvT，，，

#include<bits/stdc++.h>
using namespace std;

stack<string>stk;
void work(string s)
{
	istringstream iss(s);
	string tmp;
	while(iss>>tmp) stk.push(tmp);
}

int main()
{
	string in,tmp;
	getline(cin,in);
	int cnt=0,cur=0;
	
	for(int i=0;i<in.size();i++)
	{
		if(in[i]=='.'||in[i]==',')
		{
			work(in.substr(cur,i-cur));
			cur=i+1;
		}
	}
		// abcde,fg
		
	while(!stk.empty())	{
		tmp=stk.top();
		stk.pop();
		if(tmp=="pintia"||tmp=="Pintia")
		{
			cnt++;
		}
	}
	cout<<cnt<<endl;
	if(cnt)
	{
		if(cnt>3)
		{
			cout<<"zhen ai la"<<endl;
		}
		else cout<<"you ai"<<endl;
	}
	else cout<<"wu gan"<<endl;
}