Submission #31719114

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;

void sol(int *p,int n){
    p[1]=n+1>>1;
    for(int i=2;i<=n;i++)if(i&1)p[i]=p[1]-i/2;
    else p[i]=p[1]+i/2;
}

int n,x,p[200005];

int main(){

    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    cin>>n>>x;
    if(n&1){
        if(x*2-1==n){
            sol(p,n);
        }else{
            p[1]=x;
            sol(p+1,n-1);
            for(int i=2;i<=n;i++)p[i]+=p[i]>=x;
        }
    }else{
        if(x*2==n||(x-1)*2==n){
            sol(p,n);
            if(p[1]!=x){
                for(int i=1;i<=n;i++)p[i]=n+1-p[i];
            }
        }else{
            p[1]=x;
            sol(p+1,n-1);
            for(int i=2;i<=n;i++)p[i]+=p[i]>=x;
        }
    }
    for(int i=1;i<=n;i++)cout<<p[i]<<' ';
    cout<<'\n';

    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘void sol(int*, int)’:
./Main.cpp:5:11: warning: suggest parentheses around ‘+’ inside ‘>>’ [-Wparentheses]
    5 |     p[1]=n+1>>1;
      |          ~^~

Judge Result