Alitaulukot

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

macro_rules! slice_delta {
    ($a:expr) => {
        ($a.iter().max().unwrap()-$a.iter().min().unwrap())
    };
}

fn main() {
    let mut input = BufReader::new(std::io::stdin());
    let mut line;
    let mut split;

    line = "".to_string();
    input.read_line(&mut line).unwrap();
    split = line.split_whitespace();
    let n: usize = split.next().unwrap().parse().unwrap();
    let k: i32 = split.next().unwrap().parse().unwrap();

    line = "".to_string();
    input.read_line(&mut line).unwrap();
    // println!("{:?}",line);
    split = line.split_whitespace();
    let ints:Vec<i32> = split.map(|t|{t.parse::<i32>().unwrap()}).collect();
    // println!("{:?}",ints);

    let mut c = n;
    let mut j: usize;
    for i in 0..n {
        j = 2;
        while i+j <= n && slice_delta!(ints[i..(i+j)]) <= k {
            // println!("{i}:{j}",i=i,j=j);
            // println!("{}",slice_delta!(ints[i..(i+j)]));
            c += 1;
            j += 1;
        }
    }
    
    println!("{}",c);
}

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: TIME LIMIT EXCEEDED

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: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED