백준 1022 : 소용돌이 예쁘게 출력하기 (C++)

https://www.acmicpc.net/problem/1022

1022번: 소용돌이 예쁘게 출력하기

틀린이유 : 아무생각없이 10^8크기의 배열을 만들어 값을 채워버림

로직
규칙을 찾아 각 좌표에 맞는 값을 얻는 함수를 구현한다.
(0,0)=1^2
(1,1)=3^2
(2,2)=5^2
(n,n)=(1+n*2)^2
임을 이용한다.
범위중 가장 큰 수를 찾아 자리수를 고정하여 출력한다.

#include<bits/stdc++.h>

#include<iostream>


#define MAX 1000000007
#define INF 2000000000
#define MOD 1000000007 
using namespace std;
typedef long long ll;
typedef pair<int, int > pii;
typedef pair<long long, long long> pll;
typedef pair<double, double> pdd;

int dx[8] = { 0, -1, 0, 1, 0, 0, -1, -1 };
int dy[8] = { -1, 0, 1, 0, -1, -1, 0, 0 };

/*
int dx[8] = { -1, -1, 1, 1, -2, -2, 2, 2 };
int dy[8] = { 2, -2, 2, -2, -1, 1, -1, 1 };
*/
void init() {
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
}
ll gcd(ll a, ll b) { for (; b; a %= b, swap(a, b)); return a; }
ll lcm(ll a, ll b) { return (a * b) / gcd(a, b); }

int getValue(int x, int y) {
    int k = max(abs(x), abs(y));
    int n = 1 + 2 * k;

    if (k == x) {
        return n * n - (k - y);
    }
    else if (k == -y) {
        return n * n - (n - 1) - (k - x);
    }
    else if (k == -x) {
        return n * n - (n - 1) * 2 - (k + y);
    }
    else if (k == y) {
        return n * n - (n - 1) * 3 - (k + x);
    }
}
int main() {
    init();
    int x1, y1, x2, y2; cin >> x1 >> y1 >> x2 >> y2;
    int mx = max(getValue(x1, y1), max(getValue(x1, y2), max(getValue(x2, y1), getValue(x2, y2))));
    int cnt = 0;
    while (mx != 0) {
        cnt++;
        mx /= 10;
    }
    cout.width(cnt);
    for (int i = x1; i <= x2; i++) {
        for (int j = y1; j <= y2; j++) {
            cout.width(cnt);
            cout << getValue(i, j) << " ";
        }
        cout << "\n";
    }
}