2020牛客多校 6E.Easy Construction（构造）

题意：

给定n和k，要求构造一个长度为n的排列，
满足对于任意[1,n]内的数字i，存在长度为i的子区间，子区间的区间和%n=k

数据范围：n<=5e3，k<n

解法：

因为长度为n的区间只有一个
区间和为n*(n+1)/2
那么需要满足n*(n+1)/2%n==k才有解

1.当n为奇数时,n*(n+1)/2%n=0,即此时k必须为0
那么可以构造{n,1,n-1,2,n-2...}
这样的话,除了a[1],其他位置相邻的和都为n
奇数长度子区间把n选进去(即选a[1])
偶数长度子区间不把n选进去(即不选a[1])

2.当n为偶数时,n*(n+1)/2%n=n/2,即此时k必须为n/2
那么可以构造{n,n/2,1,n-1,2,n-2...}
奇数长度子区间把n/2选进去
偶数长度子区间把n和n/2都选进去

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=5e3+5;
int a[maxm];
int n,k;
signed main(){
    cin>>n>>k;
    if(n*(n+1)/2%n!=k){
        cout<<-1<<endl;
        return 0;
    }
    if(n%2==1){
        a[1]=n;
        int now=1;
        for(int i=3;i<=n;i+=2){
            a[i-1]=now;
            a[i]=n-now;
            now++;
        }
    }else{
        a[1]=n;
        a[2]=n/2;
        int now=1;
        for(int i=4;i<=n;i+=2){
            a[i-1]=now;
            a[i]=n-now;
            now++;
        }
    }
    for(int i=1;i<=n;i++){
        cout<<a[i]<<' ';
    }
    return 0;
}

---