結果

問題 No.217 魔方陣を作ろう
ユーザー Kmcode1Kmcode1
提出日時 2015-05-26 22:59:30
言語 C++11
(gcc 11.4.0)
結果
TLE  
実行時間 -
コード長 2,345 bytes
コンパイル時間 1,097 ms
コンパイル使用メモリ 103,392 KB
実行使用メモリ 8,768 KB
最終ジャッジ日時 2023-09-20 15:03:35
合計ジャッジ時間 7,852 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 26 ms
4,380 KB
testcase_02 TLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cctype>
#include<cstdlib>
#include<algorithm>
#include<bitset>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<sstream>
#include<fstream>
#include<iomanip>
#include<ctime>
#include<complex>
#include<functional>
#include<climits>
#include<cassert>
#include<iterator>
#include<unordered_map>
#include<unordered_set>
using namespace std;
int n;
#define MAX 21
int a[MAX][MAX];
set<int> s;
int sum = 0;
int sum1 = 0;
int sum2 = 0;
int row[MAX];
int col[MAX];
void output(){
	for (int i = 0; i < n; i++){
		for (int j = 0; j < n; j++){
			if (j){
				printf(" ");
			}
			printf("%d", a[i][j]);
		}
		puts("");
	}
	exit(0);
}
bool ok(int a, int b,int ad){
	if (row[a] + ad > sum){
		return false;
	}
	if (col[b] + ad > sum){
		return false;
	}
	if (a == b){
		if (sum1 + ad > sum){
			return false;
		}
	}
	if (a == n - b - 1){
		if (sum2 + ad > sum){
			return false;
		}
	}
	return true;
}
void add(int a, int b, int ad){
	row[a] += ad;
	col[b] += ad;
	if (a == b){
		sum1 += ad;
	}
	if (a == n - b - 1){
		sum2 += ad;
	}
}
inline void dfs(int aa, int bb){
	//cout << aa << " " << bb << endl;
	if (s.size() == 1){
		if (sum1 + (*s.begin()) == sum&&row[aa] + (*s.begin()) == sum&&col[bb] + (*s.begin()) == sum){
			a[aa][bb] = *s.begin();
			output();
		}
		return;
	}
	if (aa == n - 1){
		int v = sum - col[bb];
		if (s.count(v)){
			if (ok(aa, bb, v) == false){
				return;
			}
			a[aa][bb] = v;
			s.erase(v);
			add(aa, bb, v);
			dfs(aa, bb+1);
			add(aa, bb, -v);
			s.insert(v);
		}
		return;
	}
	if (bb == n - 1){
		int v = sum - row[aa];
		if (s.count(v)){
			if (ok(aa, bb, v) == false){
				return;
			}
			a[aa][bb] = v;
			s.erase(v);
			add(aa, bb, v);
			dfs(aa + 1, 0);
			add(aa, bb, -v);
			s.insert(v);
		}
		return;
	}
	set<int>::iterator ite;
	for (ite = s.begin(); ite != s.end(); ite++){
		if (rand() % 17 == 0){
			return;
		}
		if (ok(aa, bb, (*ite))){
			add(aa, bb,(*ite));
			int val = *ite;
			a[aa][bb] = val;
			s.erase(ite);
			dfs(aa, bb + 1);
			s.insert(val);
			ite = s.lower_bound(val);
			add(aa, bb, -val);
		}
	}
}
int main(){
	scanf("%d", &n);
	for (int i = 1; i <= n*n; i++){
		s.insert(i);
		sum += i;
	}
	sum /= n;
	dfs(0, 0);
	return 0;
}
0