Algorithm/문제

19. 점수계산

챛채 2023. 6. 14. 21:02

한국 정보 올림피아드 문제

 

변수 cnt=0 

점수 누적하는 변수 answer

1을 만나면 cnt++하고 answer에 누적

import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.Scanner;

public class Main {

    public int solution(int n, int[] arr) {
        int answer =0, cnt=0;
        for (int i = 0; i < n; i++) {
            if(arr[i] == 1) {
                cnt++;
                answer += cnt;
            }
            else cnt = 0;
        }
        return answer;
    }
   public static void main(String[] args){
       Main T = new Main();
       Scanner kb = new Scanner(System.in);
       int n=kb.nextInt(); //문제 개수 입력 받음
       int[] arr =new int [n];
       for (int i = 0; i < n; i++) {
           arr[i]=kb.nextInt();
       }
       System.out.println(T.solution(n, arr));

   }
}

쉽다..