[C++] 직사각형 좌표 구하기

문제 설명

직사각형을 만드는 데 필요한 4개의 점 중 3개의 좌표가 주어질 때, 나머지 한 점의 좌표를 구하려고 합니다. 점 3개의 좌표가 들어있는 배열 v가 매개변수로 주어질 때, 직사각형을 만드는 데 필요한 나머지 한 점의 좌표를 return 하도록 solution 함수를 완성해주세요. 단, 직사각형의 각 변은 x축, y축에 평행하며, 반드시 직사각형을 만들 수 있는 경우만 입력으로 주어집니다.

제한사항

• v는 세 점의 좌표가 들어있는 2차원 배열입니다.
• v의 각 원소는 점의 좌표를 나타내며, 좌표는 [x축 좌표, y축 좌표] 순으로 주어집니다.
• 좌표값은 1 이상 10억 이하의 자연수입니다.
• 직사각형을 만드는 데 필요한 나머지 한 점의 좌표를 [x축 좌표, y축 좌표] 순으로 담아 return 해주세요.

---

입출력 예

v	result
[[1, 4], [3, 4], [3, 10]]	[1, 10]
[[1, 1], [2, 2], [1, 2]]	[2, 1]

입출력 예 #1
세 점이 [1, 4], [3, 4], [3, 10] 위치에 있을 때, [1, 10]에 점이 위치하면 직사각형이 됩니다.

입출력 예 #2
세 점이 [1, 1], [2, 2], [1, 2] 위치에 있을 때, [2, 1]에 점이 위치하면 직사각형이 됩니다.

#include <iostream>
#include <vector>
using namespace std;

vector<int> solution(vector<vector<int> > v) {
	vector<int> ans;
	int x, y;

	ans.assign(2, 0);

	if (v[0][0] == v[1][0])
		x = v[2][0];
	else if (v[0][0] == v[2][0])
		x = v[1][0];
	else if (v[1][0] == v[2][0])
		x = v[0][0];

	if (v[0][1] == v[1][1])
		y = v[2][1];
	else if (v[0][1] == v[2][1])
		y = v[1][1];
	else if (v[1][1] == v[2][1])
		y = v[0][1];

	ans[0] = x;
	ans[1] = y;

	return ans;
}

- 좀더 정리 해보면

#include <iostream>
#include <vector>
using namespace std;

vector<int> solution(vector<vector<int> > v) {
	vector<int> ans(2,0);
	int *coordinate = new int[2];

	for (int i = 0; i < 2; i++){
		if (v[0][i] == v[1][i])
			coordinate[i] = v[2][i];
		else if (v[0][i] == v[2][i])
			coordinate[i] = v[1][i];
		else if (v[1][i] == v[2][i])
			coordinate[i] = v[0][i];
	}

	for(int j=0; j< 2; j++)
		ans[j] = coordinate[j];

	return ans;
}
void main()
{
	vector<vector<int> > problem(3, vector<int>(2,0)); // >>를 붙여쓰면 쉬프트 연산자로 인식할 수 있음
	vector<int> ans;

	problem[0][0] = 1;
	problem[0][1] = 4;
	problem[1][0] = 3;
	problem[1][1] = 4;
	problem[2][0] = 3;
	problem[2][1] = 10;

	ans = solution(problem);

	cout << "[" << ans[0] << ", " << ans[1] << "]" << endl;
	
}

- 쉬운 문제지만 벡터가 주어진 만큼 이를 사용해서 더 효율적으로 코드를 짜는 방법이 있지 않을까?

- vector container에 대해 정리할 필요성을 느꼈다.

 

지금이라면 이렇게 풀거 같은데 XOR 방식보다 더 좋아보이지는 않는다(20210909)

#include <iostream>
#include <vector>
#include <map>

using namespace std;

vector<int> solution(vector<vector<int> > v) {
    vector<int> ans;
    map<int, int> coord_x;
    map<int, int> coord_y;
    for (auto coord : v) {
        coord_x[coord[0]] == 0 ? coord_x[coord[0]]++ : coord_x[coord[0]]--;
        coord_y[coord[1]] == 0 ? coord_y[coord[1]]++ : coord_y[coord[1]]--;
    }
    
    for (auto x : coord_x) {
        if(x.second != 0) {
            ans.push_back(x.first);
        }
    }
    
    for (auto y : coord_y) {
        if(y.second != 0) {
            ans.push_back(y.first);
        }
    }
    
    return ans;
}