JZ初中OJ 1999.[2015.8.6普及組模擬賽] Wexley接蘋果

 input:apple.in output:apple.out

時間限制:  1000 ms  空間限制: 128000 KB  具體限制  

題目描述

         Wexley最近發現了一個古老的屏幕游戲。游戲的屏幕被划分成n列。在屏幕的底端，有一個寬為m列的籃子（m<n）。在游戲過程中，Wexley能左右移動這個籃子，            Wexley的操作很犀利，移動是瞬間完成的，但是籃子必須始終都在屏幕中。 蘋果從屏幕的頂端落下，每個蘋果從n列中的某一列頂端掉落，垂直掉落到屏幕的底端。每個蘋果總是在上一個蘋果掉落到底端的時候開始掉落。Wexley想要通過移動籃子來接住所有的蘋果。起先，籃子在屏幕的最左端。
         求出Wexley要接住所有的蘋果所需移動的最短距離。 

 

輸入

第一行，兩個整數n、m，如題所述
第二行，一個整數k，表示掉落的蘋果總數
接下來k行，每行一個整數Ai，表示每個蘋果掉落的位置

輸出

一行一個整數，表示所需移動最短距離

 

樣例輸入

Sample Input1:
5 1
3
1
5
3

Sample Input2:
5 2
3
1
5
3

 

樣例輸出

Sample Output1:
6

Sample Output2:
4

 

 

數據范圍限制

【數據范圍】
對於30%的數據，m<n<=5,k<=10
對於100%的數據，1<=m<n<=10,1<=k<=20

#include<bits/stdc++.h>
using namespace std;
int n,m,k,a[21];
int b[21],head,ans,tmp;
int main()
{
    freopen("apple.in","r",stdin);
    freopen("apple.out","w",stdout);
    cin>>n>>m>>k;
    b[0]=1;
    for(int i=1;i<=k;i++)
    {
        cin>>a[i];
    }
    head=m;
    for(int i=1;i<=k;i++)
    {
        if(head>=a[i] && (head-m+1)<=a[i])
        continue;
        if(head<a[i])
        {
            ans+=a[i]-head;
            head+=a[i]-head;
            continue;
        }
        if((head-m+1)>a[i])
        {
            int tmp=head;
            head=a[i]+m-1;
            ans+=tmp-head;
            continue;
        }
    }
    cout<<ans;
}