kmjp's blog

競技プログラミング参加記です

Codeforces #324 Div2 D. Dima and Lisa

色んな解き方がありそう。
http://codeforces.com/contest/584/problem/D

問題

奇数Nが与えられる。
Nを3つ以内の素数の和に分割せよ。

解法

  • Nが素数ならNを答える。
  • N-2が素数なら、2と(N-2)を答える。
  • 上のどちらも成り立たない場合、適当に小さい素数を2個Nから引いて、残りが素数か判定する。
    • 引く2個の素数を1000位総当たりすれば十分うまく行く。

最後のパターンは、最初に3固定で引くとか、Nに近い素数を探して引くなど、色々バリエーションがあるようだ

int N;

const int prime_max = 20000;
int NP,prime[100000],divp[prime_max];
map<int,int> M;

void cprime() {
	for(int i=2;i<prime_max;i++) if(divp[i]==0) {
		//M[i]=NP;
		prime[NP++]=i;
		for(ll j=1LL*i*i;j>i&&j<prime_max;j+=i) divp[j]=i;
	}
}

bool isprime(ll v) {
	for(ll i=2;i*i<=v;i++) if(v%i==0) return false;
	return true;
}

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N;
	
	if(isprime(N)) return _P("1\n%d\n",N);
	if(isprime(N-2)) return _P("2\n2 %d\n",N-2);
	
	cprime();
	FOR(x,NP) FOR(y,NP) {
		int z = N-(prime[x]+prime[y]);
		if(z>0 && isprime(z)) return _P("3\n%d %d %d\n",prime[x],prime[y],z);
	}
}

まとめ

すごい雑な回答だけど通ったからまぁいいか…?