Flat Earth (백준 22986)

안녕하세요~~ 오늘은 평평한 땅을 만들거예요.

문제

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

아이디어

지구의 크기가 N-K일때부터 N일때가지의 땅끝의 칸의 개수를 세어주면 됩니다. 이는 등차수열의 합으로 나타낼 수 있지요.

지구의 크기가 x일때의 땅끝의 칸의 개수는 4x입니다.

소스코드

#include <iostream>
#define ll long long
using namespace std;

int main(void)
{
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    ll T, N, K;
    cin >> T;
    while (T--)
    {
        cin >> N >> K;
        if (N < K) K = N;
        cout << (K + 1) * (N + N - K) * 2 << "\n";
    }
}

감사합니다.