[Baekjoon] 9037_The candy war

문제(출처: https://www.acmicpc.net/problem/9037)

---

< The candy war >

 

문제 풀이 

 

사탕의 절반을 오른쪽에 넘겨주고 홀수개라면 1을 더해주는 과정을 전체 값이 같아질 때까지 반복한다.

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

public class _9037_ { // The candy war

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;

		int T = Integer.parseInt(bf.readLine());

		for (int i = 0; i < T; i++) {
			int N = Integer.parseInt(bf.readLine());

			int cnt = 0;

			int arr[] = new int[N];
			int temp[] = new int[N];
			st = new StringTokenizer(bf.readLine());
			for (int j = 0; j < N; j++) {
				arr[j] = Integer.parseInt(st.nextToken());
				if (arr[j] % 2 != 0) {
					arr[j] += 1;
				}
			}

			boolean flag = false;
			for (int j = 0; j < N - 1; j++) {
				if (arr[j] != arr[j + 1]) {
					flag = true;
					break;
				}
			}
			if (!flag) {
				bw.write(cnt + "\n");
			} else {
				while (true) {
					flag = false;
					for (int j = 0; j < N; j++) {
						arr[j] /= 2;
						temp[j] = arr[j];
					}
					for (int j = 0; j < N; j++) {
						if (j + 1 == N) {
							arr[0] += temp[j];
							if (arr[0] % 2 != 0) {
								arr[0] += 1;
							}
						} else {
							arr[j + 1] += temp[j];
							if (arr[j + 1] % 2 != 0) {
								arr[j + 1] += 1;
							}
						}

					}
					cnt += 1;
					for (int j = 0; j < N - 1; j++) {
						if (arr[j] != arr[j + 1]) {
							flag = true;
							break;
						}
					}
					if (!flag) {
						bw.write(cnt + "\n");
						break;
					}
				}
			}
		}
		bw.flush();
	}
}

변수)
T :  테스트 케이스 개수
N : 아이의 인원수
cnt : 순환
arr : 각 아이가 가지는 사탕 수
temp : 각 아이의 사탕 절반 값
flag : 모두 같은 사탕인지 여부

 

테스트 케이스 개수를 입력받는다. 테스트 케이스 수만큼 다음 과정을 반복한다.

 

1) 아이의 인원수 입력받음

2) 아이의 인원수만큼 사탕의 개수 입력받아 arr에 저장

3) arr에 저장된 값이 홀수개라면 1 더함

4) 모든 사탕의 값이 같은 값인지 확인 후 같은 값이라면 0 출력 후 1)로 이동

5) 모든 사탕의 값이 같아질 때까지 사탕의 개수를 반으로 저장 후 절반을 오른쪽에 넘겨줌. 가진 사탕의 수가 홀수개라면 1 더함. 모든 사탕의 값이 같아졌다면 cnt 출력 후 1)로 이동

---

---