프로그래머스 - 탑

▶ 문제 설명

[ 문제 ] 

수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다.

예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7인 네 번째 탑의 신호는 높이가 9인 두 번째 탑이, 높이가 5인 세 번째 탑의 신호도 높이가 9인 두 번째 탑이 수신합니다. 높이가 9인 두 번째 탑과 높이가 6인 첫 번째 탑이 보낸 레이저 신호는 어떤 탑에서도 수신할 수 없습니다.

 

송신 탑(높이)	수신 탑(높이)
5(4)	4(7)
4(7)	2(9)
3(5)	2(9)
2(9)	-
1(6)	-

맨 왼쪽부터 순서대로 탑의 높이를 담은 배열 heights가 매개변수로 주어질 때 각 탑이 쏜 신호를 어느 탑에서 받았는지 기록한 배열을 return 하도록 solution 함수를 작성해주세요.

제한 사항

• heights는 길이 2 이상 100 이하인 정수 배열입니다.
• 모든 탑의 높이는 1 이상 100 이하입니다.
• 신호를 수신하는 탑이 없으면 0으로 표시합니다.

[ 입출력 예 ] 

heights	return
[6,9,5,7,4]	[0,0,2,2,4]
[3,9,9,3,5,7,2]	[0,0,0,3,3,3,6]
[1,5,3,6,7,6,5]	[0,0,2,0,0,5,6]

 

---

오늘 문제는 lv2이지만 쉬운 편이었습니다.

요즘 알고리즘을 풀 때, 사람이 생각하는것처럼 짜려고 생각하는 편인데 이번 문제는 사람이 풀 때에도 이중포문을 돌아야 하기 때문에 딱히 머리를 쓸 필요가 없었네요 ..

 

코드를 짤 때에 어려웠던 점은 없었지만,

처음에 풀었을 때에는 포문내에 bool변수를 두고 있는지 체크하는 부분을 생성했는데 생각해보니 answer[i]를 계속 0으로 초기화를 해 주면 코드가 훨씬 간단해질 것 같아 변경하였습니다. 

 

 

[ 처음 짰을 때 답변 ] 

 

// heights_len은 배열 heights의 길이입니다.
int* solution(int heights[], size_t heights_len) {
	// return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
	int* answer = (int*)malloc(sizeof(int)*heights_len);

	for (int i = (heights_len - 1) ; 0 < i; --i)
	{
		bool isfind = false;

		for (int j = (i - 1) ; 0 <= j ; --j )
		{
			if (heights[i] < heights[j])
			{
				answer[i] = j+1;
				isfind = true;
				break;
			}
		}
		if (false == isfind)
		{
			answer[i] = 0;
		}
	}
	answer[0] = 0;

	return answer;
}

 

 

[ 간략화 시킨 답변 ] 

 

// heights_len은 배열 heights의 길이입니다.

int* solution(int heights[], size_t heights_len) 
{
	// return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
	int* answer = (int*)malloc(sizeof(int)*heights_len);

	for (int i = (heights_len - 1) ; 0 <= i; --i)
	{
		answer[i] = 0;
		for (int j = (i - 1) ; 0 <= j ; --j )
		{
			if (heights[i] < heights[j])
			{
				answer[i] = j+1;
				break;
			}
		}
	}
	return answer;
}

 

 

 

 

 

알고리즘을 짤 때에 최대한 합칠 수 있는 라인은 합치려고 노력하고 있는데 여러번 풀어서 간결해진 코드를 보면 뿌듯한 느낌이 드네요 :)

앞으로 더 간결하게 풀어보려고 합니다.