SSLOJ·Wexley接苹果【模拟】

Description–

Wexley最近发现了一个古老的屏幕游戏。
游戏的屏幕被划分成n列。在屏幕的底端，有一个宽为m列的篮子（m<n）。在游戏过程中，Wexley能左右移动这个篮子，
Wexley的操作很犀利，移动是瞬间完成的，但是篮子必须始终都在屏幕中。
苹果从屏幕的顶端落下，每个苹果从n列中的某一列顶端掉落，垂直掉落到屏幕的底端。
每个苹果总是在上一个苹果掉落到底端的时候开始掉落。
Wexley想要通过移动篮子来接住所有的苹果。
起先，篮子在屏幕的最左端。
求出Wexley要接住所有的苹果所需移动的最短距离。

---

Input–

第一行，两个整数n、m，如题所述
第二行，一个整数k，表示掉落的苹果总数
接下来k行，每行一个整数Ai，表示每个苹果掉落的位置

Output–

一行一个整数，表示所需移动最短距离

---

Sample Input–

Sample Input1:

5 1
3
1
5
3

Sample Input2:

5 2
3
1
5
3

Sample Output–

Sample Output1:

6

Sample Output2:

4

---

代码–

#include<iostream>
#include<cstdio>

using namespace std;

int a, n, m, l, r, k, ans;

int main()
{
	scanf("%d%d%d", &n,&m,&k);
	l = 1, r = m;
	for (int i = 1; i <= k; ++i)
	{
		scanf("%d", &a);
		if (l <= a && r >= a) continue;
		if (l > a) ans += l - a, l = a, r = a + m - 1;//在篮子左边
		if (r < a) ans += a - r, l = a - m + 1, r = a;//在篮子右边
	}
	printf("%d", ans);
	
	return 0;
}