[알고리즘] Java / 백준 / 기차가 어둠을 헤치고 은하수를 / 15787
[알고리즘] Java / 백준 / 기차가 어둠을 헤치고 은하수를 / 15787
문제
접근 방식
열차 하나 당 20개의 좌석이 있으므로 비트로 열차 내 정보를 표현하면 한 열차 당 20비트이다.
따라서 int 배열을 만들고 비트 마스킹으로 각 명령어를 수행하여 열차 정보를 갱신한다
그 후 집합에 정수형인 모든 열차 정보를 집어넣어 중복을 제거한 후 집합의 길이를 출력한다
코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.StringTokenizer;
public class Main_15787 {
static int[] trains;
public static void main(String[] args) throws IOException {
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());
// 기차 배열 (인덱스 1 부터 시작, N까지)
trains = new int[N+1];
// 명령어 수행
for(int i=0;i<M;i++) {
excute(br.readLine());
}
// 중복 제거
HashSet<Integer> set = new HashSet<>();
for(int i=1;i<N+1;i++) {
set.add(trains[i]);
}
// 출력
System.out.println(set.size());
}
public static void excute(String command) {
StringTokenizer st = new StringTokenizer(command);
int c = Integer.parseInt(st.nextToken()); // 명령
int i = Integer.parseInt(st.nextToken()); // 기차 번호
int x = -1; // 좌석 번호 ( x가 1부터 시작해서 0부터로 바꿔주기 위해 초기값 -1로 설정, 이후 x에 입력값을 더하는 방식)
switch (c){
// i번째 기차 x번째 좌석에 사람 태우기
case 1:
x += Integer.parseInt(st.nextToken());
trains[i] |= (1 << x);
break;
// i번째 기차 x번째 좌석의 사람 하차시키기
case 2:
x += Integer.parseInt(st.nextToken());
trains[i] &= ~(1 << x);
break;
// i번째 기차 한 칸씩 뒤로 밀고 맨 뒤 사람 하차시키기
case 3:
trains[i] = (trains[i] & ~(1 << 19)) << 1;
break;
// i번째 기차 맨 앞 사람 하차시키고 한 칸씩 앞으로 당기기
case 4:
trains[i] = (trains[i] & ~(1 << 0)) >> 1;
break;
}
}
}
Author And Source
이 문제에 관하여([알고리즘] Java / 백준 / 기차가 어둠을 헤치고 은하수를 / 15787), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@gandi0330/알고리즘-Java-백준-기차가-어둠을-헤치고-은하수를-15787저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)