[코드트리/NOVICE MID] 특정 구간의 원소 평균값 (JAVA)

🔺 문제

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

 

🔺 코드

import java.util.*;
import java.io.*;
 
public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        int[] arr = new int[N];
 
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        for(int i = 0 ; i < N ; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }
 
        int answer = 0;
 
        for(int i = 0 ; i < N ; i++) {  // 구간 시작점
            for(int j = i ; j < N ; j++) {  // 구간 끝점
                boolean exists = false;
                double tmpSum = 0;
                double avg = 0;
 
                for(int k = i ; k <= j ; k++) {  // 구간 내 합 계산하기
                    tmpSum += arr[k];
                }
                
                avg = tmpSum / (j - i + 1); // 평균 구하기
 
                for(int k = i ; k <= j ; k++) {  // 해당 구간 안에 평균값과 같은 값이 있는지 확인
                    if((double)arr[k] == avg)
                        exists = true;
                }
 
                if(exists)
                    answer++;
            }            
        }
 
        System.out.println(answer);
    }
}

 

 

 

🧩  해결 아이디어

• 완전탐색

- 구간 시작점 정하기 (i : 0 ~ N-1까지)

-  구간 끝점 정하기 (j : i ~ N-1까지)

-  구간 내 합 구하고, 구간 안에 평균값과 같은 값이 있는지 확인 (k : i ~j까지)

- 값이 있다면, 정답 + 1

 

 

 

---

💬 느낀 점

빠르게 풉시다....ㅠ

 

1회독	2회독	3회독	4회독	5회독
V

---

(참고)

[코드트리] 특정 구간의 원소 평균값 / 완전 탐색