题目内容

球迷排成一队换购世界杯徽章。换购规则如下：

• 从队头开始，依次为每人换购 $1$ 枚徽章，每次换购耗时 $1$ 个单位时间；
• 若该球迷仍有未换完的徽章，则重新排到队尾继续等待；
• 若该球迷所有徽章已换完，则直接离队，不再参与后续排队。

给定每人需要换购的徽章数量数组 medals 和目标球迷下标 $k$，求当数组下标第 $k$ 号球迷换完所有徽章时，总耗时为多少？

输入描述

medals：长度为 $n$ 的正整数数组，medals[i] 表示数组下标第 $i$ 号球迷需要换购的徽章数；

$k$：目标球迷所在的数组下标，输入保证 $0 \le k < n$；

输出描述

输出一个整数，表示当数组下标第 $k$ 号球迷换完所有徽章时的总耗时。

样例1

输入

[2, 3, 2], 2

输出

6

说明

初始队伍：[0号(2), 1号(3), 2号(2)]，其中 $2$ 号是目标球迷。

• 第 $1$ 轮遍历：

  • $0$ 号换 $1$ 枚，剩余 $1$，排到队尾，耗时 $+1$（累计 $1$）
  • $1$ 号换 $1$ 枚，剩余 $2$，排到队尾，耗时 $+1$（累计 $2$）
  • $2$ 号换 $1$ 枚，剩余 $1$，排到队尾，耗时 $+1$（累计 $3$）

  队伍变为：[0号(1), 1号(2), 2号(1)]。

• 第 $2$ 轮遍历：

  • $0$ 号换 $1$ 枚，剩余 $0$，离队，耗时 $+1$（累计 $4$）
  • $1$ 号换 $1$ 枚，剩余 $1$，排到队尾，耗时 $+1$（累计 $5$）
  • $2$ 号换 $1$ 枚，剩余 $0$，换完离队，耗时 $+1$（累计 $6$）

$2$ 号球迷已换完所有徽章，总耗时 $= 6$。

样例2

输入

[1, 5, 3], 1

输出

9

说明

数组下标 $1$ 号球迷需要换购 $5$ 个徽章，第 $1$ 轮结束 $0$ 号离队（累计 $3$）；第 $3$ 轮结束 $2$ 号离队（累计 $3+2+2=7$）；第 $5$ 轮结束 $1$ 号全部换购（累计 $3+2+2+1+1=9$）。