본문 바로가기
Problem Solving/Baekjoon

[백준] 2891 카약과 강풍 - Greedy / Java

by graycode 2022. 12. 28.

 문제 링크

 

2891번: 카약과 강풍

첫째 줄에 팀의 수 N, 카약이 손상된 팀의 수 S, 카약을 하나 더 가져온 팀의 수 R이 주어진다. (2 ≤ N ≤ 10, 1 ≤ S, R ≤ N) 둘째 줄에는 카약이 손상된 팀의 번호가 주어진다. 팀 번호는 중복되지 않

www.acmicpc.net

 

 풀이 과정

 

 풀이 코드

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

public class Main {

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

        int n = Integer.parseInt(st.nextToken());
        int s = Integer.parseInt(st.nextToken());
        int r = Integer.parseInt(st.nextToken());

        int[] arr = new int[n + 2];
        Arrays.fill(arr, 1);

        st = new StringTokenizer(br.readLine());
        while (s-- > 0) arr[Integer.parseInt(st.nextToken())]--;

        st = new StringTokenizer(br.readLine());
        while (r-- > 0) arr[Integer.parseInt(st.nextToken())]++;

        for (int i = 1; i <= n; i++) {
            if (arr[i] == 2) {
                if (arr[i - 1] == 0) arr[i - 1]++;
                else if (arr[i + 1] == 0) arr[i + 1]++;
            }
        }

        int cnt = 0;
        for (int i = 1; i <= n; i++)
            if (arr[i] == 0) cnt++;

        bw.write(String.valueOf(cnt));
        bw.flush();
    }

}

댓글