Apartments

#![allow(unused)]

use std::{io, usize};

fn take_tuple() -> (usize, usize, usize) {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let mut it = input
        .trim()
        .split_whitespace()
        .map(|x| x.parse::<usize>().unwrap());
    (it.next().unwrap(), it.next().unwrap(), it.next().unwrap())
}

fn take_vector() -> Vec<usize> {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let arr: Vec<usize> = input
        .trim()
        .split_whitespace()
        .map(|x| x.parse::<usize>().unwrap())
        .collect();
    return arr;
}

fn main() {
    let (num_applicants, num_apartments, allowed_diff): (usize, usize, usize) = take_tuple();
    let mut applicants: Vec<usize> = take_vector();
    applicants.sort_unstable();
    let mut apartments: Vec<usize> = take_vector();
    apartments.sort_unstable();

    let mut count: usize = 0;
    let mut applicant_index: usize = 0;
    let mut apartments_index: usize = 0;

    while applicant_index < num_applicants && apartments_index < num_apartments {
        if apartments[apartments_index] < applicants[applicant_index] - allowed_diff {
            apartments_index += 1;
        } else if apartments[apartments_index] > applicants[applicant_index] + allowed_diff {
            applicant_index += 1;
            continue;
        } else {
            count += 1;
            apartments_index += 1;
            applicant_index += 1;
        }
    }
    println!("{}", count);
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: ACCEPTED

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: WRONG ANSWER

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: WRONG ANSWER

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED