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));
}
}
쉽다..