[Java] 수열의 경우수

- 바이토닉 수열의 개수 반환

 

(입력)

1, 3, 2, 5, 7, 4, 2, 5, 1

 

(출력)

 

 

# ArrayList 사용

import java.util.ArrayList;

public class Main {

    private int solution(int[] nums) {

        int result = 0;
        int n = nums.length;
        ArrayList<Integer> peaks = new ArrayList<>();

        for(int i=1; i<n-1; i++) {
            if(nums[i-1] < nums[i] && nums[i] > nums[i+1])
                peaks.add(i);
        }

        for(int p : peaks) {
            int lt = p;
            int rt = p;
            int lcnt = 0;
            int rcnt = 0;

            while(lt-1 >= 0 && nums[lt-1] < nums[lt]) {
                lt--;
                lcnt++;
            }
            while(rt+1 < n && nums[rt] > nums[rt+1]) {
                rt++;
                rcnt++;
            }
            result += (lcnt*rcnt);
        }

        return result;
    }

    public static void main(String[] args) {
        Main T = new Main();
        int[] nums = new int[]{1, 3, 2, 5, 7, 4, 2, 5, 1};
        System.out.println(T.solution(nums)); // 6
    }
}