SWEA 8016. 홀수 피라미드 (long 범위를 넘어가는 BigInteger)

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AWvzGUKKPVwDFASy&categoryId=AWvzGUKKPVwDFASy&categoryType=CODE&problemTitle=&orderBy=INQUERY_COUNT&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=11

SW Expert Academy

 

재밌는 문제이다. 개발하면서 쓰지도 않아본 BigInteger를 코테하면서 쓸 줄이야.

 

BigInteger는 클래스라서 사칙연산에 대해 메서드 호출로 해야된다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.math.BigInteger;

class Main {
    static int T;
    static BigInteger N;

    public static void main(String args[]) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        T = Integer.parseInt(br.readLine());
        for (int tc = 1; tc <= T; tc++) {
            N = new BigInteger(br.readLine());

            BigInteger two = BigInteger.valueOf(2);
            BigInteger one = BigInteger.ONE;

            BigInteger termN = N.multiply(N); // N * N
            BigInteger termN1 = termN.subtract(N.multiply(two).subtract(one)); // N^2 - (2N - 1)
            BigInteger first = termN1.multiply(two).subtract(one).add(two); // (2 * termN1 - 1) + 2
            BigInteger second = termN.multiply(two).subtract(one); // (2 * termN - 1)

            System.out.printf("#%d %s %s\n", tc, first.toString(), second.toString());
        }
    }
}

계차수열로 풀 수 있다는데 수학 공부를 하는게 아니니까 스킵