USACO Training Section1.4 Arithmetic Progressions

一个很简单的遍历，居然递交了N次才通过。差不多能犯的错都犯了，比如，memory.h没有include（在VS中编译通过，OJ编译不通过）；没有满足要求的答案时要输出"NONE"，一开始忘记了；做了个小的边界条件优化结果错了一位；最后补充边界条件检查（cur<n_nums）时直接提交结果少了个s。前两题实际上比这个难做点（写的程序复杂些）都是一次过，看来真是大意失荆州啊。

题目的时限是5秒，看了看结果，用了3.9秒。基本上没做任何优化，可以考虑：
1.加上取消重复数列的判断，比如，1 12 13 12的后者可以跳过。
2.输出时可以不用排序，用一个指针存公差相等序列，由于搜索是有序的，保存的结果也有序的，就可以直接输出。
我的程序完全暴力而没TLE，主要因为内循环中使用指针和register变量。

原题链接：
[url]http://ace.delos.com/usacoprob2?a=1N7165lJnJE&S=ariprog[/url]

/*
ID: blackco3
TASK: ariprog
LANG: C++
*/
#include <fstream>
#include <iostream>
#include <stdlib.h>
#include <memory.h>
using namespace std ;

int N, M ;
#define _max_M_ 250
int is_bisquare[_max_M_*_max_M_*2+1] ;
int bisquares[(_max_M_+1)*(_max_M_+1)/2], n_bisquare ;
#define _max_seq_ 10000
struct t_seq {
	int start, dif ;
} seqs[_max_seq_];
int seq_comp( const void *s1, const void *s2 ) {
	return ((t_seq*)s1)->dif==((t_seq*)s2)->dif ?
		((t_seq*)s1)->start-((t_seq*)s2)->start : ((t_seq*)s1)->dif-((t_seq*)s2)->dif ;
}

int main() {
    ofstream fout ("ariprog.out");
    ifstream fin ("ariprog.in");
	fin >> N >> M ;
	int n_nums=((M*M)<<1)+1 ;
	memset( is_bisquare, 0, sizeof(int)*n_nums );
	for( int p=0; p<=M; p++ )
		for( int q=p; q<=M; q++ )
			is_bisquare[ p*p + q*q ]=1 ;
	n_bisquare = 0 ;
	for( int i=0; i<n_nums; i++ )
		if( is_bisquare[i] )
			bisquares[ n_bisquare++ ]=i;
	int *pc, *pn, *p_end=bisquares+n_bisquare-N+1 ;
	t_seq *p_seq=seqs ;
	for( pc=bisquares; pc!=p_end; pc++ ){
		for( pn=pc+1; pn!=p_end+1; pn++ ){
			register int dif=*pn-*pc, n_dif=2, cur=*pn+dif ;
			while( cur<n_nums && is_bisquare[cur] && n_dif!=N ){
				cur+=dif, n_dif++ ;
			}
			if( n_dif==N )
				p_seq->start=*pc, p_seq->dif=dif, p_seq++ ;  
		}
	}
	int n_seq = p_seq-seqs ;
	if( !n_seq ) {
		fout << "NONE" << endl ;
	} else {
		qsort( seqs, n_seq, sizeof(t_seq), seq_comp );
		for( p_seq=seqs; p_seq!=seqs+n_seq; p_seq++ )
			fout << p_seq->start << " " << p_seq->dif << endl ; 
	}
	return 0;
}