점수 계산

문제

나의풀이

import java.util.*;

class Main {
	public int solution(int n,int[] arr) {
			int answer = 0,count = 0;
			for(int i=0; i<n; i++) {
				//맞은경우, 1
				if(arr[i] == 1) {
					if(count == 0) {
						answer += 1;
						count++;
					}
					else {
						answer += count+1;
						count++;
					}
				}
				//틀린경우, 0
				else {
					count = 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));
	}
	
}

더 간단한 풀이

	public int solution(int n,int[] arr) {
			int answer = 0,count = 0;
			for(int i=0; i<n; i++) {
				//맞은경우, 1
				if(arr[i] == 1) {
					count += 1;
					answer += count;
				}
				//틀린경우, 0
				else {
					count = 0;
				}
			}
			return answer;
		}

풀이방법

1. 내가 처음 푼것은 카운트가 0일때와 아닐때를 나눠서 생각했는데,
   틀린것은 아니지만 조금만 더 생각하면 더 짧게 나타낼수있다.

2. answer에 1을더하고 count를 따로 더하는것이 아니라 count에 먼저 누적을시키고 count를 answer에 더하는것으로 해결할 수 있다.

핵심키워드

간단해 보이는 문제도 더 효율적으로 푸는 방법을 생각하자!