한국 정보 올림피아드 문제
변수 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));
}
}
쉽다..
'Algorithm > 문제' 카테고리의 다른 글
21. 격자판 최대합 (1) | 2023.06.22 |
---|---|
20. 등수 구하기 (0) | 2023.06.22 |
18. 뒤집은 소수 ★ (0) | 2023.06.12 |
17. 소수(에라토스테네스의 체) (0) | 2023.06.12 |
16. 피보나치 수열 (0) | 2023.06.01 |