#531540 (C++14) No.223 1マス指定の魔方陣

提出ソース

結果

問題	No.223 1マス指定の魔方陣
ユーザー	heno239
提出日時	2020-08-15 21:37:07
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	2 ms / 5,000 ms
コード長	5,140 bytes
コンパイル時間	1,762 ms
コンパイル使用メモリ	133,492 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-04-19 08:21:09
合計ジャッジ時間	3,518 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,812 KB
testcase_02	AC	1 ms 6,816 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	2 ms 6,940 KB
testcase_05	AC	2 ms 6,944 KB
testcase_06	AC	2 ms 6,944 KB
testcase_07	AC	2 ms 6,944 KB
testcase_08	AC	2 ms 6,940 KB
testcase_09	AC	2 ms 6,944 KB
testcase_10	AC	2 ms 6,940 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	2 ms 6,940 KB
testcase_13	AC	2 ms 6,940 KB
testcase_14	AC	2 ms 6,940 KB
testcase_15	AC	2 ms 6,940 KB
testcase_16	AC	2 ms 6,940 KB
testcase_17	AC	2 ms 6,940 KB
testcase_18	AC	2 ms 6,944 KB
testcase_19	AC	2 ms 6,940 KB
testcase_20	AC	2 ms 6,944 KB
testcase_21	AC	2 ms 6,940 KB
testcase_22	AC	2 ms 6,944 KB
testcase_23	AC	2 ms 6,940 KB
testcase_24	AC	2 ms 6,940 KB
testcase_25	AC	2 ms 6,944 KB
testcase_26	AC	2 ms 6,944 KB
testcase_27	AC	2 ms 6,944 KB
testcase_28	AC	2 ms 6,940 KB
testcase_29	AC	2 ms 6,940 KB
testcase_30	AC	2 ms 6,940 KB
testcase_31	AC	2 ms 6,940 KB
testcase_32	AC	2 ms 6,940 KB
testcase_33	AC	2 ms 6,944 KB
testcase_34	AC	2 ms 6,940 KB
testcase_35	AC	2 ms 6,944 KB
testcase_36	AC	2 ms 6,944 KB
testcase_37	AC	2 ms 6,944 KB
testcase_38	AC	2 ms 6,940 KB
testcase_39	AC	2 ms 6,940 KB
testcase_40	AC	2 ms 6,940 KB
testcase_41	AC	2 ms 6,944 KB
testcase_42	AC	2 ms 6,940 KB
testcase_43	AC	2 ms 6,940 KB
testcase_44	AC	2 ms 6,940 KB
testcase_45	AC	2 ms 6,940 KB
testcase_46	AC	2 ms 6,940 KB

権限があれば一括ダウンロードができます

コンパイルメッセージ

main.cpp: In function 'void solve()':
main.cpp:201:31: warning: 's' may be used uninitialized [-Wmaybe-uninitialized]
  201 |                 while (p[goa] != s) {
main.cpp:184:21: note: 's' was declared here
  184 |                 int s, g;
      |                     ^
main.cpp:184:24: warning: 'g' may be used uninitialized [-Wmaybe-uninitialized]
  184 |                 int s, g;
      |                        ^

ソースコード

raw source code

#include<iostream>
#include<string>
#include<cstdio>
#include<vector>
#include<cmath>
#include<algorithm>
#include<functional>
#include<iomanip>
#include<queue>
#include<ciso646>
#include<random>
#include<map>
#include<set>
#include<bitset>
#include<stack>
#include<unordered_map>
#include<utility>
#include<cassert>
#include<complex>
#include<numeric>
using namespace std;

//#define int long long
typedef long long ll;

typedef unsigned long long ul;
typedef unsigned int ui;
constexpr ll mod = 1000000007;
const ll INF = mod * mod;
typedef pair<int, int>P;
#define stop char nyaa;cin>>nyaa;
#define rep(i,n) for(int i=0;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define Rep(i,sta,n) for(int i=sta;i<n;i++)
#define rep1(i,n) for(int i=1;i<=n;i++)
#define per1(i,n) for(int i=n;i>=1;i--)
#define Rep1(i,sta,n) for(int i=sta;i<=n;i++)
#define all(v) (v).begin(),(v).end()
typedef pair<ll, ll> LP;
typedef double ld;
typedef pair<ld, ld> LDP;
const ld eps = 1e-12;
const ld pi = acos(-1.0);

ll mod_pow(ll x, ll n, ll m = mod) {
	ll res = 1;
	while (n) {
		if (n & 1)res = res * x % m;
		x = x * x % m; n >>= 1;
	}
	return res;
}
struct modint {
	ll n;
	modint() :n(0) { ; }
	modint(ll m) :n(m) {
		if (n >= mod)n %= mod;
		else if (n < 0)n = (n % mod + mod) % mod;
	}
	operator int() { return n; }
};
bool operator==(modint a, modint b) { return a.n == b.n; }
modint operator+=(modint& a, modint b) { a.n += b.n; if (a.n >= mod)a.n -= mod; return a; }
modint operator-=(modint& a, modint b) { a.n -= b.n; if (a.n < 0)a.n += mod; return a; }
modint operator*=(modint& a, modint b) { a.n = ((ll)a.n * b.n) % mod; return a; }
modint operator+(modint a, modint b) { return a += b; }
modint operator-(modint a, modint b) { return a -= b; }
modint operator*(modint a, modint b) { return a *= b; }
modint operator^(modint a, ll n) {
	if (n == 0)return modint(1);
	modint res = (a * a) ^ (n / 2);
	if (n % 2)res = res * a;
	return res;
}

ll inv(ll a, ll p) {
	return (a == 1 ? 1 : (1 - p * inv(p % a, a)) / a + p);
}
modint operator/(modint a, modint b) { return a * modint(inv(b, mod)); }

ll gcd(ll a, ll b) {
	if (a < b)swap(a, b);
	while (b) {
		ll r = a % b; a = b; b = r;
	}
	return a;
}

vector<vector<int>> ad[3] = { {{4,1},{2,3}},{{1,4},{2,3}},{{1,4},{3,2}} };
vector<vector<int>> make_s(int n) {
	if (n % 2) {
		vector<vector<int>> res(n, vector<int>(n));
		vector<vector<bool>> used(n, vector<bool>(n, false));
		int x = 0, y = n / 2;
		rep1(i, n * n) {
			res[x][y] = i; used[x][y] = true;
			if (i < n * n) {
				int nx = x - 1, ny = y + 1;
				if (nx < 0)nx += n;
				if (ny == n)ny = 0;
				if (used[nx][ny]) {
					nx = x + 1, ny = y;
					if (nx == n)nx = 0;
				}
				x = nx, y = ny;
			}
		}
		return res;
	}
	if (n % 4 == 0) {
		vector<vector<int>> res(n, vector<int>(n));
		vector<vector<int>> col(n, vector<int>(n, 0));

		rep(i, n / 4)rep(j, n / 4) {
			rep(k, 4) {
				col[i * 4 + k][j * 4 + k] = 1;
				col[i * 4 + k][j * 4 + 3 - k] = 1;
			}
		}
		rep(i, n)rep(j, n) {
			if (col[i][j]) {
				res[i][j] = i * n + j + 1;
			}
		}
		per(i, n)per(j, n) {
			if (!col[i][j]) {
				res[i][j] = (n - 1 - i) * n + (n - 1 - j) + 1;
			}
		}
		return res;
	}
	int d = n / 2;
	vector<vector<int>> res(n, vector<int>(n));
	vector<vector<int>> p = make_s(d);
	rep(i, d)rep(j, d) {
		p[i][j] = (p[i][j] - 1) * 4;
	}
	rep(i, d)rep(j, d) {
		int lx = 2 * i, ly = 2 * j;
		int col = 0;
		if (i > d / 2 + 1)col = 2;
		else if (i < d / 2 + 1) {
			col = 0;
			if (i == d / 2 && j == d / 2)col = 1;
		}
		else {
			col = 1;
			if (j == d / 2)col = 0;
		}
		rep(s, 2)rep(t, 2) {
			res[lx + s][ly + t] = p[i][j] + ad[col][s][t];
		}
	}
	return res;
}
//clockwise
void sousa(vector<vector<int>> &v, int b) {
	int n = v.size();
	rep(i, n / b/2)rep(j, n / b/2) {
		vector<vector<int>> memo[2][2];
		rep(x, 2)rep(y, 2) {
			memo[x][y].resize(b, vector<int>(b));
			rep(dx, b)rep(dy, b) {
				memo[x][y][dx][dy] = v[i * 2 * b + x * b + dx][j * 2 * b + y * b + dy];
			}
		}
		swap(memo[0][0], memo[0][1]);
		swap(memo[0][0], memo[1][0]);
		swap(memo[1][0], memo[1][1]);
		rep(x, 2)rep(y, 2) {
			rep(dx, b)rep(dy, b) {
				v[i * 2 * b + x * b + dx][j * 2 * b + y * b + dy] = memo[x][y][dx][dy];
			}
		}
	}
}
void solve() {
	int n; cin >> n;
	vector<vector<int>> ans = make_s(n);
	int y, x, val;
	cin >> y >> x >> val; y--; x--;
	int lx = 0, ly = 0;
	for (int b = n / 2; b >= 1; b >>= 1) {
		int s, g;
		rep(dx, 2)rep(dy, 2)rep(i, b)rep(j, b) {
			int cx = lx + dx * b + i;
			int cy = ly + dy * b + j;
			if (cx == x && cy == y) {
				s = dx * 2 + dy;
			}
			if (ans[cx][cy] == val) {
				g = dx * 2 + dy;
			}
		}
		vector<int> p = { 0,1,3,2,0,1,3,2 };
		int goa = 0;
		rep(i, 4) {
			if (p[i] == g)goa = i;
		}
		int cnt = 0;
		while (p[goa] != s) {
			cnt++; goa++;
		}
		rep(i, cnt)sousa(ans, b);
		if (s & 2)lx += b;
		if (s & 1)ly += b;
	}
	rep(i, n) {
		rep(j, n) {
			if (j > 0)cout << " ";
			cout << ans[i][j];
		}
		cout << "\n";
	}
}





signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	//cout << fixed << setprecision(15);
	//init_f();
	//init();
	//expr();
	//int t; cin >> t; rep(i, t)
	solve();
	return 0;
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード