1
impl Solution {
2
  pub fn time_required_to_buy(mut tickets: Vec<i32>, k: i32) -> i32 {
3
    let mut sum = 0i32;
4
    let mut N = tickets[k as usize];
5
    for i in 0..tickets.len() {
6
      if N > tickets[i] {
7
        sum += tickets[i]
8
      } else {
9
        sum += N
10
      }
11
      if i == k as usize {
12
        N -= 1
13
      }
14
    }
15
    return sum;
16
  }
17
}