USACO1.4 Arithmetic Progressions(ariprog)

最新推荐文章于 2024-05-06 00:08:33 发布

weixin_33898876

最新推荐文章于 2024-05-06 00:08:33 发布

阅读量73

点赞数 1

文章标签： python c/c++ 数据结构与算法

原文链接：https://my.oschina.net/u/347565/blog/62199

版权

2019独角兽企业重金招聘Python工程师标准>>>

使用暴力搜索，枚举b,a的所有情况，进行检测，注意剪枝。首先是使用数组bool x[i]保存i值是否满足i==p*p+q*q这一条件，这一使用空间换时间的方法比使用检测函数快得多。第二个是注意b和a的取值范围，b是[1，2*m*m/(n-1)]，a是[0,2*m*m-(n-1)*b]。控制取值范围方法一：在for循环第二个表达式中控制好进入循环条件；方法二：先进入for循环再求值检测，不满足要求就跳出循环。使用方法二时第8组测试例超时半秒，用方法一只用方法二的一半时间，可见应该尽量控制好进入循环的条件而不是先进入再检测。

/*
ID:jzzlee1
PROG:ariprog
LANG:C++
*/
#include<iostream>
#include<fstream>
#include<cstring>
#include<cctype>
#include<cstdio>
using namespace std;
ifstream fin("ariprog.in");
ofstream fout("ariprog.out");
int n,m,s[26],x[1300000];
int main()
{
	int a,b,i,mm;
	fin>>n>>m;
	//cin>>n>>m;
	 for(int p=0;p<=m;p++)
        for(int q=0;q<=m;q++)
            x[p*p+q*q]=1;
	mm=2*m*m;bool flag=1,sign=0;
	for(b=1;b<=mm/(n-1);b++)
	{
		
		for(a=0;a<=mm-(n-1)*b;a++)
		{
			flag=1;
			for(i=0;i<n;i++)
			{
				s[i]=a+i*b;
				if(!x[s[i]])
				{
					flag=0;
					break;
				}
			}
			if(flag)
			{
				//cout<<a<<" "<<b<<endl;
			fout<<a<<" "<<b<<endl;
			sign=1;
			}
		}
	}
	if(!sign)
	//cout<<"NONE"<<endl;
	fout<<"NONE"<<endl;
	return 0;
}

转载于:https://my.oschina.net/u/347565/blog/62199