Method func

Codechef/Java/RedAlert.java:21–40

(int[] arr, int n, int d, int h)

    }

    public static void func(int[] arr, int n, int d, int h) {
        int waterLevel = 0;
        for (int i = 0; i < n; i++) {
            if (arr[i] > 0) {
                waterLevel += arr[i];
            } else if (arr[i] == 0) {
                if (waterLevel <= d) {
                    waterLevel = 0;
                } else {
                    waterLevel -= d;
                }

            }
            if (waterLevel > h) {
                System.out.println("YES");
                return;
            }
        }
        System.out.println("NO");
    }
}