Alitaulukot

warning: variable does not need to be mutable
  --> input/code.rs:25:9
   |
25 |     let mut ints:Vec<i32> = split.map(|t|{t.parse::<i32>().unwrap()}).collect();
   |         ----^^^^
   |         |
   |         help: remove this `mut`
   |
   = note: `#[warn(unused_mut)]` on by default

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

macro_rules! c_comp {
    ($a:expr,$b:expr,$c:expr) => {
        $a < $b && $b < $c
    };
}

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: i32 = 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 mut ints:Vec<i32> = split.map(|t|{t.parse::<i32>().unwrap()}).collect();
    // println!("{:?}",ints);

    let mut c = n;
    let mut j; 
    let mut note:HashSet<[i32;2]> = HashSet::new();
    for i in 0..n {
        // println!("{}",i);
        j = 1;
        while i+j < n && c_comp!(-1,-ints[(i+j) as usize]+ints[i as usize],k+1) {
            // println!("{:?}",note);
            if note.contains(&[i,i+j]) {break;}
            // println!("{}:{}",i,j);
            note.insert([i,i+j]);
            c += 1;
            j += 1;
        }

        j = 1;
        while i-j >= 0 && c_comp!(-1,-ints[(i-j) as usize]+ints[i as usize],k+1) {
            if note.contains(&[i-j,i]) {break;}
            // println!("{}:-{}",i,j);
            note.insert([i-j,i]);
            c += 1;
            j += 1;
        }
    }
    println!("{}",c);
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 5

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 2, 3

Verdict: WRONG ANSWER

Test 8

Group: 2, 3

Verdict: WRONG ANSWER

Test 9

Group: 2, 3

Verdict: WRONG ANSWER

Test 10

Group: 2, 3

Verdict: WRONG ANSWER

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: WRONG ANSWER

Test 13

Group: 3

Verdict: WRONG ANSWER

Test 14

Group: 3

Verdict: WRONG ANSWER

Test 15

Group: 3

Verdict: WRONG ANSWER

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED