1008. 数组元素循环右移问题

题目截图：

思路：

　　先将整个数组逆置，然后将数组前 M 个元素和后面的元素分别逆置即可。

代码：

/*
    1008. 数组元素循环右移问题
*/

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <time.h>

#define maxn 101
int p[maxn] = {0};

// 逆置 
void swap(int a, int b) {
    int i;
    for(i=a; i<=(a+b)/2; ++i) {
        int temp = p[i];
        p[i] = p[a+b-i];
        p[a+b-i] = temp;
    }
}

int main() {
    int N, M, i, flag=0;
    scanf("%d %d", &N, &M);
    M %= N;                    // M 可能大于 N 
    for(i=0; i<N; ++i) {
        scanf("%d", &p[i]);
    }
    if(M != 0) {            // M 为 0 不需要移动 
        swap(0, N-1);        // 逆置整个数组 
        swap(0, M-1);        // 逆置数组前 M 个元素 
        swap(M, N-1);        // 逆置数组后面元素 
    }
    for(i=0; i<N; ++i) {    // 按格式输出 
        if(flag) {
            printf(" ");
        }
        printf("%d", p[i]);
        flag = 1;
    }

    return 0;
}