백준 17225 세훈이의 선물가게

노문택·2022년 4월 4일
0
post-thumbnail

https://www.acmicpc.net/problem/17225

큐 문제를 많이 풀어보지못해서 이것도 남의 풀이를 확인해버렷다 ㅠㅠ..

Queue 문제이기보다는 PriorityQueue 좀더 근접한 문제이다..

예제입력과 예시가 잘나와서..

중복일때는 지수를 먼저! 라고 생각하면 된다..

풀이는 다음과같다.

  1. PQ를 만들고 그안에서 순서를 정렬해줄 CLASS를 만들어주고

  2. 그클래스는 시작시간 // 상자종류로 구성을한다.

  3. 입력으로 오는 시간과 작업시작 시간이 같지않을수도있으니까 예외처리해주면 끝

그래서 코드는 다음과같다..

import java.util.*;
import java.io.*;
public class 세훈이의선물가게 {

	static class Product implements Comparable<Product>{
		
		int time;
		int type; //A는 1 B는 2로줌
		
		public Product(int time, int type) {
			this.time= time;
			this.type= type;
		}
		public int compareTo(Product o) {
			if(this.time==o.time) {
				return this.type-o.type;
			}
			return this.time-o.time;
		}
	}
	public static void main(String[] args) throws Exception {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());
		int C = Integer.parseInt(st.nextToken());
		
		int Amax = -1;
		int Bmax = -1;
		int Acount = 0;
		int Bcount= 0;
		PriorityQueue<Product> pq = new PriorityQueue<>();
		for(int i=0;i<C;i++) {
			
			st = new StringTokenizer(br.readLine());
			
			int start = Integer.parseInt(st.nextToken());
			String type = st.nextToken();
			int count = Integer.parseInt(st.nextToken());
			
			int ntype=1;
			if(type.equals("R")) { //지수작업
				ntype=2;
				Bcount+= count;
				if(Bmax<start) Bmax = start;
				for(int j=0;j<count;j++) {
					pq.add(new Product(Bmax,ntype));
					Bmax+=M;
				}
				
			}
			else { //세훈작업
				Acount+= count;
				if(Amax<start) Amax = start;
				for(int j=0;j<count;j++) {
					pq.add(new Product(Amax,ntype));
					Amax+=N;
				}
			}
		}
		
		
		int index = 1;
		StringBuilder asb = new StringBuilder();
		asb.append(Acount).append("\n");
		StringBuilder bsb = new StringBuilder();
		bsb.append(Bcount).append("\n");
		while(!pq.isEmpty()) {
			Product cur = pq.poll();
			if(cur.type==2) {
				bsb.append(index).append(" ");
				index++;
			}
			else {
				asb.append(index).append(" ");
				index++;
			}

		}
		System.out.println(asb);
		System.out.println(bsb);
		
		
	}

}

너무너무어렵다..큐문제..일단 골드5로 뚜벅뚜벅 ㄱㄱ

profile
노력하는 뚠뚠이

0개의 댓글