[코드트리] 배열의 특정 위치 찾기 3 - 자바(Java)

문제

https://www.codetree.ai/missions/4/problems/find-specific-location-fo-array-3/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

 

코드

최대 100개의 정수를 입력받아 처음으로 입력받은 0의 앞에 있는 3개의 정수들의 합을 출력하는 문제다.

• 0은 네 번째 이후부터 주어진다.

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int[] arr = new int[100];
        int sum = 0;

        for (int i = 0; i < arr.length; i++) {
            int n = Integer.parseInt(st.nextToken());
            
            if (n == 0) {
                sum = arr[i - 1] + arr[i - 2] + arr[i - 3];
                break;
            }
            
            arr[i] = n;
        }

        System.out.println(sum);
    }
}