[programmers] 롤케이크 자르기

문제
https://school.programmers.co.kr/learn/courses/30/lessons/132265

프로그래머스

 

---

< 롤케이크 자르기 >

 

문제 풀이 (Java)

class Solution {
	public int solution(int[] topping) {
		int answer = 0;

		int arr1[] = new int[10001];
		int arr2[] = new int[10001];

		int cnt1 = 0;
		int cnt2 = 0;

		for (int i = 0; i < topping.length; i++) {
			if (arr1[topping[i]] == 0) {
				cnt1 += 1;
			}
			arr1[topping[i]] += 1;
		}

		for (int i = 0; i < topping.length; i++) {
			arr1[topping[i]] -= 1;
			if (arr1[topping[i]] == 0) {
				cnt1 -= 1;
			}

			if (arr2[topping[i]] == 0) {
				cnt2 += 1;
			}
			arr2[topping[i]] += 1;

			if (cnt1 == cnt2) {
				answer += 1;
			}
		}
		return answer;
	}
}

 

일단 롤케이크 자르기 전 전체를 철수꺼라 생각하고 토핑 수를 센다. 그 후 앞에서부터 토핑 하나씩 동생한테 주면서 토핑 수가 같을 때의 수를 센다. 

 

---

---

 

출처: 프로그래머스 코딩 테스트 연습, 
https://school.programmers.co.kr/learn/challenges