[Baekjoon] 15465_Milk Measurement

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

---

< Milk Measurement >

 

문제 풀이 

 

ArrayList에 소 이름, 현재 우유 생산량을 저장하며 생산량이 변경된 각 날마다 우유 생산량을 기준으로 정렬하여 사진 바뀐 여부를 구한다.

 

my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.StringTokenizer;

public class _15465_ { // Milk Measurement
	static class Cow {
		private String name;
		private int milk;
		private int day;

		public Cow(String name, int milk) {
			this.name = name;
			this.milk = milk;
		}

		public Cow(String name, int milk, int day) {
			this.name = name;
			this.milk = milk;
			this.day = day;
		}
	}

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

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

		ArrayList<Cow> list = new ArrayList<>();
		list.add(new Cow("Bessie", 7));
		list.add(new Cow("Elsie", 7));
		list.add(new Cow("Mildred", 7));

		ArrayList<Cow> cmd = new ArrayList<>();
		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(bf.readLine());

			int day = Integer.parseInt(st.nextToken());
			String name = st.nextToken();
			int milk = Integer.parseInt(st.nextToken());

			cmd.add(new Cow(name, milk, day));
		}

		Collections.sort(cmd, new Comparator<Cow>() {
			@Override
			public int compare(Cow o1, Cow o2) {
				return o1.day - o2.day;
			}
		});

		int result = 0;
		HashSet<String> set = new HashSet<>();
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < 3; j++) {
				if (list.get(j).name.equals(cmd.get(i).name)) {
					list.get(j).milk += cmd.get(i).milk;
					break;
				}
			}
			Collections.sort(list, new Comparator<Cow>() {
				@Override
				public int compare(Cow o1, Cow o2) {
					return o2.milk - o1.milk;
				}
			});

			boolean flag = false;
			for (int j = 0; j < 3; j++) {
				if (list.get(j).milk == list.get(0).milk) {
					if (!set.contains(list.get(j).name)) {
						flag = true;
						set.add(list.get(j).name);
					}
				} else {
					if (set.contains(list.get(j).name)) {
						flag = true;
						set.remove(list.get(j).name);
					}
				}
			}
			if (flag) {
				result += 1;
			}
		}
		System.out.println(result);
	}
}

변수)
n : 측정 횟수
list : 각 소에 대해 우유 생산량
cmd : 측정 기록
day, name, milk : 날짜, 소의 이름, 우유 생산량의 변화량
result : 벽에 걸어둔 소의 사진을 바꿔야 했던 날의 수 

 

측정 횟수를 입력받는다. ArrayList에 초기 각 3마리의 소에 대해 이름과 초기 우유 생산량 7을 저장한다. 측정 횟수만큼 측정값을 입력받아 새로운 ArrayList에 저장한다. 측정값을 저장한 ArrayList를 날짜 기준 오름차순으로 정렬한다. 측정값을 살펴보며 소의 우유 생산량을 변화시킨다. 소 3마리의 정보를 저장한 ArrayList를 우유 생산량 기준 내림차순으로 정렬한 후 벽에 걸어둔 소의 사진을 바꿔야 하는지 판별한다. 바꿔야 한다면 result를 +1 한다.

 

최종 result를 출력한다. 

---

---