2020牛客暑假多校第六场 E-Easy Construction
题意:
给定n和k,要求构造一个长度为n的排列,
满足对于任意[1,n]内的数字i,存在长度为i的子区间,子区间的区间和%n=k
题解:
因为长度为n的区间只有一个,判断((1+n)*n/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 LL long long
int main(){
int n,k;
cin>>n>>k;
int sum=n*(n+1)/2;
if(sum%n!=k%n)cout<<-1<<endl;
else if(n&1){
cout<<n;
for(int i=1;i<=n/2;i++)cout<<" "<<i<<" "<<n-i;
cout<<endl;
}else{
cout<<n<<" "<<n/2;
for(int i=1;i<n/2;i++)cout<<" "<<i<<" "<<n-i;
cout<<endl;
}
system("pause");
return 0;
}