結果
| 問題 | No.223 1マス指定の魔方陣 |
| ユーザー |
 koyumeishi
|
| 提出日時 | 2015-06-05 23:59:48 |
| 言語 | C++11 (gcc 11.4.0) |
| 結果 |
CE
(最新)
AC
(最初)
|
| 実行時間 | - |
| コード長 | 4,403 bytes |
| コンパイル時間 | 550 ms |
| コンパイル使用メモリ | 77,880 KB |
| 最終ジャッジ日時 | 2024-11-14 19:04:08 |
| 合計ジャッジ時間 | 918 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge1 |
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
コンパイルメッセージ
main.cpp: In function ‘std::vector<std::vector<int> > Magic_Square(int)’:
main.cpp:43:9: error: ‘iota’ was not declared in this scope
43 | iota(tmp.begin(), tmp.end(), 1);
| ^~~~
main.cpp: In function ‘bool check(std::vector<std::vector<int> >&)’:
main.cpp:102:15: error: ‘accumulate’ was not declared in this scope
102 | int sum = accumulate(v[0].begin(), v[0].end(),0);
| ^~~~~~~~~~
main.cpp: In function ‘int main()’:
main.cpp:202:16: error: ‘function’ was not declared in this scope
202 | vector<function<void()>> func = {swap_a, swap_b, swap_e, swap_f};
| ^~~~~~~~
main.cpp:13:1: note: ‘std::function’ is defined in header ‘<functional>’; did you forget to ‘#include <functional>’?
12 | #include "assert.h"
+++ |+#include <functional>
13 | using namespace std;
main.cpp:202:30: error: template argument 1 is invalid
202 | vector<function<void()>> func = {swap_a, swap_b, swap_e, swap_f};
| ^
main.cpp:202:30: error: template argument 2 is invalid
main.cpp:202:31: error: expected unqualified-id before ‘>’ token
202 | vector<function<void()>> func = {swap_a, swap_b, swap_e, swap_f};
| ^~
main.cpp:207:17: error: ‘func’ was not declared in this scope
207 | func[a]();
| ^~~~
ソースコード
#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <set>
#include <ctime>
#include "assert.h"
using namespace std;
vector<vector<int>> Magic_Square(int n){
vector<vector<int>> v(n, vector<int>(n,0));
if(n&1){ //odd
int x = n/2;
int y = 0;
int cnt = 0;
v[y][x] = ++cnt;
while(cnt < n*n){
x = (x+1)%n;
y = (y-1+n)%n;
if(v[y][x] != 0){
x = (x-1+n)%n;
y = (y+2)%n;
}
v[y][x] = ++cnt;
}
}else if(n%4 == 0){ //4k
int x = 0;
int y = 0;
for(int i=0; i<n; i++){
for(int k=0; k<n/4; k++){
v[i][(i+k*4)%n] = -1;
v[i][(3+k*4-i+n)%n] = -1;
}
}
vector<int> tmp(n*n);
iota(tmp.begin(), tmp.end(), 1);
set<int> s(tmp.begin(), tmp.end());
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
if(v[i][j] < 0){
v[i][j] = i*n + j + 1;
s.erase(v[i][j]);
}
}
}
for(int i=n-1; i>=0; i--){
for(int j=n-1; j>=0; j--){
if(v[i][j] == 0){
v[i][j] = *s.begin();
s.erase(v[i][j]);
}
}
}
}else{ //4k+2
auto u = Magic_Square((n-2)/2 + 1);
for(int i=0; i<u.size(); i++){
for(int j=0; j<u.size(); j++){
u[i][j] = (u[i][j] - 1)*4;
}
}
vector<vector<int>> w((n-2)/2+1, vector<int>((n-2)/2 + 1, 0));
for(int i=0; i<w.size(); i++){
w[w.size()/2 + 1][i] = 1;
}
swap(w[w.size()/2 + 1][w.size()/2], w[w.size()/2][w.size()/2]);
for(int i=w.size()/2 + 2; i<w.size(); i++){
for(int j=0; j<w.size(); j++){
w[i][j] = 2;
}
}
int z[3][2][2] = {
{{4,1},{2,3}},
{{1,4},{2,3}},
{{1,4},{3,2}}
};
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
v[i][j] = u[i/2][j/2];
int w_ = w[i/2][j/2];
v[i][j] += z[w_][i&1][j&1];
}
}
}
return v;
}
bool check(vector<vector<int>>& v){
int sum = accumulate(v[0].begin(), v[0].end(),0);
for(int i=0; i<v.size(); i++){
if(sum != accumulate(v[i].begin(), v[i].end(), 0)) return false;
int sum_r = 0;
for(int j=0; j<v.size(); j++){
sum_r += v[j][i];
}
if(sum != sum_r) return false;
}
int sum_ = 0;
for(int i=0; i<v.size(); i++){
sum_ += v[i][i];
}
if(sum_ != sum) return false;
sum_ = 0;
for(int i=0; i<v.size(); i++){
sum_ += v[v.size()-1-i][v.size()-1-i];
}
if(sum_ != sum) return false;
return true;
}
int main(){
int n,x,y,z;
cin >> n >> x >> y >> z;
x--;y--;
auto v = Magic_Square(n);
int val = v[y][x];
auto print = [&](){
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
cerr << v[i][j] << " ";
}cerr << endl;
}
cerr << endl;
};
print();
auto swap_a = [&](){
for(int k=1; k+1<n; k+=4){
for(int i=0; i<n; i++){
swap(v[i][k], v[i][k+1]);
}
}
for(int k=1; k+1<n; k+=4){
for(int i=0; i<n; i++){
swap(v[k][i], v[k+1][i]);
}
}
};
auto swap_b = [&](){
for(int k=0; k<n; k+=4){
for(int i=0; i<n; i++){
swap(v[i][k], v[i][k+1]);
}
}
for(int k=0; k<n; k+=4){
for(int i=0; i<n; i++){
swap(v[k][i], v[k+1][i]);
}
}
};
auto swap_e = [&](){
auto tmp = v;
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
v[i][j] = tmp[(i+1)%n][(j+1)%n];
}
}
};
auto swap_f = [&](){
auto tmp = v;
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
v[i][j] = tmp[(i-1+n)%n][(j+1)%n];
}
}
};
srand((unsigned)time(NULL));
vector<function<void()>> func = {swap_a, swap_b, swap_e, swap_f};
while(v[y][x] != z){
auto last = v;
int a;
a = rand()%4;
func[a]();
//if(check(v) == false) v = last;
}
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
cout << v[i][j] << " ";
}cout << endl;
}
assert(check(v));
//assert(v[y][x] == z);
return 0;
}