AOJ 1189 Prime Caves

こうゆうのをパッと書けるのはいいですね。

問題

Prime Caves | Aizu Online Judge

感想

やります、以上。なんつって。

素数洞穴と素数判定表を作ったのちDPすればよさそう

ただ、どう再現すればいいのかというところにだいぶ苦労する感じです

素数洞穴は大きい順に右下からぐるぐる回っていくように作りました。

素数表はエラトスナントカの何とかですね。

DPは

dp[i][j] := 洞穴の上からi番目、左からj番目のところに来た時、最大何個素数を通れていて、最後の素数がいくつか

ですね、pairでまとめるとそのままmaxで処理できます。

DPの遷移については簡単な二重ループで出来るとうれしいなとか思うと、最初だけマークを付けておいて、マークがついているところのみ処理するみたいなことできれいにできます。(日本語ひどくないですか)

こうゆう問題をサラッと解けるとうれしくなります。

コード

#include <bits/stdc++.h>

using namespace std;

#ifdef DEBUG_MODE
	#define DBG(n) n;
#else
	#define DBG(n) ;
#endif
#define REP(i,n) for(ll (i) = (0);(i) < (n);++i)
#define rep(i,s,g) for(ll (i) = (s);(i) < (g);++i)
#define rrep(i,s,g) for(ll (i) = (s);i >= (g);--(i))
#define PB push_back
#define MP make_pair
#define FI first
#define SE second
#define SHOW1d(v,n) {for(int W = 0;W < (n);W++)cerr << v[W] << ' ';cerr << endl << endl;}
#define SHOW2d(v,i,j) {for(int aaa = 0;aaa < i;aaa++){for(int bbb = 0;bbb < j;bbb++)cerr << v[aaa][bbb] << ' ';cerr << endl;}cerr << endl;}
#define ALL(v) v.begin(),v.end()
#define Decimal fixed<<setprecision(20)
#define INF 1000000000
#define MOD 1000000007
#define M_MAX 1111

typedef long long ll;
typedef pair<ll,ll> P;

int xx[M_MAX*M_MAX];
int yy[M_MAX*M_MAX];
pair<int,int> dp[M_MAX][M_MAX];
bool isp[M_MAX*M_MAX];
int mp[M_MAX][M_MAX];
int dx[] = {-1,0,1,0};
int dy[] = {0,-1,0,1};

void f(){
	int x = M_MAX-1;
	int y = M_MAX-1;
	int now = M_MAX*M_MAX;
	int cou = 0;
	
	while(now > 0){
		mp[y][x] = now;
		xx[now] = x;
		yy[now] = y;
		now--;
		int tmpx = x + dx[cou];
		int tmpy = y + dy[cou];
		
		if(tmpx >= 0 && tmpx < M_MAX && tmpy >= 0 && tmpy < M_MAX && 
		mp[tmpy][tmpx] == 0){
			x = tmpx;
			y = tmpy;
		}
		else{
			cou = (cou + 1) % 4;
			x += dx[cou];
			y += dy[cou];
		}
	}
	
	REP(i,M_MAX*M_MAX)isp[i] = true;
	isp[0] = isp[1] = false;
	for(int i = 2;i < M_MAX*M_MAX;i++){
		if(isp[i]){
			for(int j = 2*i;j < M_MAX*M_MAX;j += i)isp[j] = false;
		}
	}
}

int main(){
	
	f();
	
	int n,m;
	
	while(cin >> m >> n,n|m){
		
		REP(i,M_MAX)REP(j,M_MAX)dp[i][j] = MP(-1,-1);
		
		dp[yy[n]][xx[n]] = MP(0,0);
		
		REP(i,M_MAX){
			REP(j,M_MAX){
				if(dp[i][j] != MP(-1,-1) && mp[i][j] <= m){
					if(isp[mp[i][j]]){
						dp[i][j] = MP(dp[i][j].FI+1,mp[i][j]);
					}
					REP(k,3){
						int x = j + dx[k];
						int y = i + 1;
						DBG(cout << "xy : " << x << ' ' << y << endl;)
						if(x >= 0 && x < M_MAX && y >= 0 && y < M_MAX){
							dp[y][x] = max(dp[i][j],dp[y][x]);
						}
					}
				}
			}
		}
		
		pair<int,int> ans = MP(0,0);
		
		REP(i,M_MAX){
			REP(j,M_MAX){
				if(mp[i][j] <= m)ans = max(ans,dp[i][j]);
			}
		}
		
		cout << ans.FI << ' ' << ans.SE << endl;
	}

	return 0;
}