CodeForces 287 D.Shifting（机智）

Description
对于一个1~n的排列p=p1p2…pn，f是一个变换：

求f(f( … f(p = [1, 2, …, n], 2) … , n - 1), n)
Input
一个整数n(2<=n<=1e6)
Output
输出f(f( … f(p = [1, 2, …, n], 2) … , n - 1), n)的结果
Sample Input
4
Sample Output
4 2 3 1
Solution
注意到做f(p,k)时，其实是把p[1]和p[k+1]交换，然后p[1]和p[2k+1]交换，然后p[1]和p[3k+1]交换，……，最后把p[1]放在序列最后，而其他位置的值只是需要右移一位，所以我们为了不让这些数动我们可以让序列在数组中存储位置的起点右移一位，这样做一次f(p,k)变换只需要进行n/k次交换，时间复杂度O(nlogn)
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 2222222
int n,a[maxn];
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)a[i]=i;
        for(int k=2;k<=n;k++)
        {
            for(int i=k-1+k;i<=n+k-2;i+=k)swap(a[k-1],a[i]);
            a[n+k-1]=a[k-1];
        }
        for(int i=n;i<2*n;i++)printf("%d%c",a[i],i==2*n-1?'\n':' ');
    }
    return 0;
}