USACO4.4.1 Shuttle Puzzle (shuttle)

不知道谁想到找规律的。
4 35 642 1357 642 35 4 (n=3)

5 46 753 2468 97531 2468 753 46 5 (n=4)

就是2*n+1组等差数列。
先求出1到n+1组，然后根据对称性求出剩下的。

奇数组为降序，偶数组为升序；奇数组首项为n+i，偶数组首项为n-i+2。

/*
ID:xsy97051
LANG:C++
TASK:shuttle
*/
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
 
int main()
{
  	freopen("shuttle.in","r",stdin);
	freopen("shuttle.out","w",stdout);

	int n,t,s,res[500],tt=0;
    memset(res,0,sizeof(res));
    cin>>n;
    for(int i=2;i<=n+1;i++)
    {
        if(i%2==1) 
            t=-2;
        else 
            t=2;
        if(i%2==1) 
            s=n+i;
        else 
            s=n-i+2;

        res[++tt]=s;
        for(int j=2;j<=i;j++)
        {
            s+=t;
            res[++tt]=s;
        }
    }
    for(int i=n;i>=2;i--)
    {
        if(i%2==1)
            t=-2;
        else 
            t=2;
        if(i%2==1) 
            s=n+i;
        else 
            s=n-i+2;

        res[++tt]=s;
        for(int j=2;j<=i;j++)
        {
            s+=t;
            res[++tt]=s;
        }
    }
    res[++tt]=n+1;
    for(int i=1;i<=tt;i++)
    {
        cout<<res[i];
        if((i%20==0)||(i==tt)) cout<<endl;
        else cout<<" ";
    }
    return 0;
}