#![allow(non_snake_case)] #[allow(unused_imports)] use proconio::{fastout, input, marker::*}; #[fastout] fn main() { input! { N: usize, S: usize, P: [usize; N], } let mut bit = DynamicBinaryIndexedTree::new(1 << 32); for &p in P.iter() { bit.add(p + S + 10, 1); } let ans = (0..N) .filter(|&i| bit.sum(P[i] + S + 10 - S..=P[i] + S + 10 + S) == 1) .collect::>(); println!( "{}\n{}", ans.len(), ans.iter() .map(|x| (x + 1).to_string()) .collect::>() .join(" ") ); } pub struct DynamicBinaryIndexedTree { size: usize, tree: std::collections::HashMap, } impl< T: Default + Clone + Copy + std::ops::Add + std::ops::AddAssign + std::ops::Sub, > DynamicBinaryIndexedTree { /// self = [0; size] pub fn new(size: usize) -> Self { return Self { size: size, tree: std::collections::HashMap::new(), }; } fn _inner_add(&mut self, mut i: usize, w: T) { assert!(i <= self.size && i > 0); while i <= self.size { let prev = self.tree.get(&i); match prev { Some(&v) => { self.tree.insert(i, v + w); } None => { self.tree.insert(i, w); } } i += i & i.wrapping_neg(); } } fn _inner_sum(&self, mut i: usize) -> T { let mut ret = T::default(); while i > 0 { if let Some(&v) = self.tree.get(&i) { ret += v; } i -= i & i.wrapping_neg(); } return ret; } /// self[i] += w pub fn add(&mut self, i: usize, w: T) { self._inner_add(i + 1, w); } /// return Σ_{j ∈ [0, i]} self[j] pub fn prefix_sum(&self, i: usize) -> T { self._inner_sum(i + 1) } /// return Σ_{j ∈ range} self[j] pub fn sum>(&self, range: R) -> T { let left = match range.start_bound() { std::ops::Bound::Included(&l) => l, std::ops::Bound::Excluded(&l) => l + 1, std::ops::Bound::Unbounded => 0, }; let right = match range.end_bound() { std::ops::Bound::Included(&r) => r, std::ops::Bound::Excluded(&r) => r - 1, std::ops::Bound::Unbounded => unreachable!(), }; if left == 0 { return self.prefix_sum(right); } else { return self.prefix_sum(right) - self.prefix_sum(left - 1); } } }