Codeforces 427E Police Patrol

找中间的数，然后从两头取。

#include<stdio.h>
const int MAX = ;
int pos[MAX];
int main()
{
    int n,m,tmp;
    int i;
    int pol;
    long long ans,dis;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        for(i=;i<=n;i++)
            scanf("%d",&pos[i]);
        pol=n/+;
        tmp=m;
        ans=;
        for(i=;i<=pol;)
        {
            dis=;
            while(i<=pol&&tmp--)
            {
                if(!dis)dis=pos[pol]-pos[i];
                i++;
            }
            ans=ans+dis*;
            tmp=m;
        }
        for(i=n;i>pol;)
        {
            dis=;
            while(i>pol&&tmp--)
            {
                if(!dis)dis=pos[i]-pos[pol];
                i--;
            }
            ans=ans+dis*;
            tmp=m;
        }
        printf("%I64d\n",ans);
    }
    return ;
}

更加简洁的代码

1<<n 就是把1向左移N位

m>>=1 相当于m= m>>1，就是把m左移1位

#include <stdio.h>
using namespace std;
const int MAX_N =  + ;
int n, m;
int arr[MAX_N];
int main()
{
    scanf("%d %d", &n, &m);
    for(int i = ; i < n; i++)
        scanf("%d", &arr[i]);
    int mid = n >> ;
    //printf("mid:%d\n",mid);
    long long res = ;
    for(int i = n - ; i > mid; i -= m)
        res += arr[i] - arr[mid];
    for(int i = ; i < mid; i += m)
        res += arr[mid] - arr[i];
    res = res << ;
    printf("%I64d\n", res);
    return ;
}