#[macro_export] macro_rules! setup { { mut $input:ident: SplitWhitespace $(,)? } => { use std::io::Read; let mut buf = String::new(); std::io::stdin().read_to_string(&mut buf).ok(); let mut $input = buf.split_whitespace(); }; } #[macro_export] macro_rules! parse_next { ($str_iter:expr) => { $str_iter.next().unwrap().parse().ok().unwrap() }; } #[allow(unused_imports)] use std::collections::{HashMap, HashSet, VecDeque}; fn main() { setup! { mut input: SplitWhitespace }; let n: usize = parse_next!(input); let k: usize = parse_next!(input); let x: Vec = (0..n).map(|_| parse_next!(input)).collect(); let a: Vec = (0..n).map(|_| parse_next!(input)).collect(); let ans = (|| { let mut queue = VecDeque::new(); let mut visited = HashSet::new(); { let root = k - 1; queue.push_back(root); visited.insert(root); } while !queue.is_empty() { let i = queue.pop_front().unwrap(); let lb = { let mut left = 0; let mut right = n; while right - left > 1 { let mid = (left + right) / 2; match x[i] - a[i] <= x[mid] { true => right = mid, false => left = mid, } } match i { i if i == 0 => 0, _ => right, } }; let ub = { let mut left = 0; let mut right = n; while right - left > 1 { let mid = (left + right) / 2; match x[mid] <= x[i] + a[i] { true => left = mid, false => right = mid, } } match i { i if i == (n - 1) => n - 1, _ => left, } }; for j in lb..=ub { if visited.contains(&j) { continue; } queue.push_back(j); visited.insert(j); } } visited.len() })(); println!("{}", ans); }