Alitaulukot

use std::io::{self, BufRead};

fn main() {
    let mut lines = io::BufReader::new(io::stdin()).lines();

    let line1 = lines.next().unwrap().unwrap();
    let line2 = lines.next().unwrap().unwrap();
    let mut line1 = line1.split_whitespace();
    let mut line2 = line2.split_whitespace();

    let n: i64 = line1.next().unwrap().parse().unwrap();
    let k: i64 = line1.next().unwrap().parse().unwrap();

    let mut arr = Vec::<i64>::new();
    for _ in 0..n {
        arr.push(line2.next().unwrap().parse().unwrap());
    }

    println!("{}", f(&arr, k));
}

fn f(arr: &[i64], k: i64) -> i64 {
    if arr.len() < 2 {
        return arr.len() as i64;
    }

    if arr.len() == 2 {
        if (arr[0] - arr[1]).abs() <= k {
            return 3;
        } else {
            return 2;
        }
    }

    let max = arr.iter().enumerate().max_by(|x, y| x.1.cmp(y.1)).unwrap();

    let mut left = 0;
    for i in (0..max.0).rev() {
        if max.1 - arr[i] > k {
            break;
        }
        left += 1;
    }

    let mut right = 0;
    for i in max.0+1..arr.len() {
        if max.1 - arr[i] > k {
            break;
        }
        right += 1;
    }

    (left + 1) * (right + 1) + f(&arr[0..max.0], k) + f(&arr[max.0+1..arr.len()], k)
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 2, 3

Verdict: ACCEPTED

Test 8

Group: 2, 3

Verdict: ACCEPTED

Test 9

Group: 2, 3

Verdict: ACCEPTED

Test 10

Group: 2, 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED