til/Algorithm
[Array] 점수계산
값진
2021. 7. 20. 16:11
import java.util.*;
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.print(T.solution(n, arr));
}
}