結果

問題 No.217 魔方陣を作ろう
ユーザー Kmcode1Kmcode1
提出日時 2015-05-26 22:54:25
言語 C++11
(gcc 13.3.0)
結果
TLE  
実行時間 -
コード長 2,305 bytes
コンパイル時間 1,696 ms
コンパイル使用メモリ 104,460 KB
実行使用メモリ 10,012 KB
最終ジャッジ日時 2024-07-06 10:08:17
合計ジャッジ時間 7,301 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 2 TLE * 1 -- * 15
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:133:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  133 |         scanf("%d", &n);
      |         ~~~~~^~~~~~~~~~

ソースコード

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 (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