[프로그래머스][C++][LV2]당구 연습

https://school.programmers.co.kr/learn/courses/30/lessons/169198

프로그래머스

 

 

 

각 좌표에 따른 조건을 나눈 다음 거리값 계산을 피타고라스를 통해 하면 된다.

거리값의 계산을 수직거리의 제곱과 수평거리의 제곱을 더해주면 된다.

#include <cmath>
#include <vector>
#include <climits>

using namespace std;
int calc_dist(int m, int n, int a, int b, int c, int d)
{
    int total = INT_MAX;
    //수평거리제곱 + 수직거리제곱
    //현재 공보다 위에 있을때
    if (a != c || b <= d) total = min(total, (int)(pow(a - c, 2) + pow(b + d, 2)));
    //현재 공보다 오른쪽에 있을 때
    if (a >= c || b != d) total = min(total, (int)(pow(a - 2 * m + c, 2) + pow(b - d, 2)));
    //현재 공보다 아래에 있을 때
    if (a != c || b >= d) total = min(total, (int)(pow(a - c, 2) + pow(b - 2 * n + d, 2)));
    //현재 공보다 왼쪽에 있을때
    if (a <= c || b != d) total = min(total, (int)(pow(a + c, 2) + pow(b - d, 2)));
    return total;
}
vector<int> solution(int m, int n, int startX, int startY, vector<vector<int>> balls)
{
    vector<int> answer;

    for (int i = 0; i < balls.size(); i++)
        answer.push_back(calc_dist(m, n, startX, startY, balls[i][0], balls[i][1]));

    return answer;
}