Submission #6,136

2023-11-03 01:04:09

cpp17

Accepted 40/40

14ms

5884 KiB

#include <bits/stdc++.h>
using namespace std;
int lk,ln,n,k,med;
vector<int> ans;
int cnt;
void solve1(int i){
        if (cnt%2==0){
            if (lk==med) lk--;
            ans[i]=lk;
            lk--;
        }
        else{
            if (ln==med) ln++;
            ans[i]=ln;
            ln++;
        }
    }

void solve2(int i){
   {
        if (cnt%2==0){
            if (ln==med) ln++;
            ans[i]=ln;
            ln++;
        }
        else{
            if (lk==med) lk--;
            ans[i]=lk;
            lk--;
        }
    }
}
int main()
{
    n,k;
    cin>>n>>k;
    ans.resize(n+1,-1);
    med=(n+1)/2;
    lk=med-1,ln=med+1;
    ans[k]=med;
    cnt=0;
    if (k%2==0){
        for (int i=k-1;i>0;i--){
            solve2(i);
            cnt++;
        }
        cnt=0;
        for (int i=k+1;i<=n;i++){
            solve1(i);
            cnt++;

        }
    }
    else{
        for (int i=k-1;i>0;i--){

            solve1(i);
            cnt++;
        }
        for (int i=k+1;i<=n;i++){
            solve2(i);
            cnt++;
        }
    }

    for (int i=1; i<=n;i++){
        cout<<ans[i]<<" ";
    }
}

	Sum	Test	Verdict		Time	Memory
base	40/40
		1	Accepted	0/0	3ms	1876 KiB
		2	Accepted	0/0	14ms	2968 KiB
		3	Accepted	1/1	3ms	2372 KiB
		4	Accepted	1/1	3ms	2556 KiB
		5	Accepted	1/1	3ms	2636 KiB
		6	Accepted	1/1	3ms	2644 KiB
		7	Accepted	1/1	3ms	2800 KiB
		8	Accepted	1/1	3ms	3052 KiB
		9	Accepted	1/1	3ms	3268 KiB
		10	Accepted	1/1	3ms	3484 KiB
		11	Accepted	1/1	2ms	3576 KiB
		12	Accepted	1/1	2ms	3536 KiB
		13	Accepted	1/1	3ms	3792 KiB
		14	Accepted	1/1	3ms	3924 KiB
		15	Accepted	1/1	3ms	4000 KiB
		16	Accepted	1/1	3ms	4116 KiB
		17	Accepted	1/1	3ms	4008 KiB
		18	Accepted	1/1	3ms	4144 KiB
		19	Accepted	1/1	3ms	4252 KiB
		20	Accepted	1/1	3ms	4156 KiB
		21	Accepted	1/1	14ms	4860 KiB
		22	Accepted	1/1	14ms	4832 KiB
		23	Accepted	1/1	14ms	4848 KiB
		24	Accepted	1/1	14ms	4852 KiB
		25	Accepted	1/1	14ms	4996 KiB
		26	Accepted	1/1	14ms	4856 KiB
		27	Accepted	1/1	14ms	4860 KiB
		28	Accepted	1/1	14ms	4868 KiB
		29	Accepted	1/1	14ms	5000 KiB
		30	Accepted	1/1	14ms	5084 KiB
		31	Accepted	1/1	14ms	5088 KiB
		32	Accepted	1/1	14ms	5176 KiB
		33	Accepted	1/1	14ms	5316 KiB
		34	Accepted	2/2	14ms	5416 KiB
		35	Accepted	1/1	14ms	5400 KiB
		36	Accepted	1/1	14ms	5320 KiB
		37	Accepted	2/2	14ms	5328 KiB
		38	Accepted	1/1	14ms	5584 KiB
		39	Accepted	1/1	14ms	5484 KiB
		40	Accepted	1/1	14ms	5884 KiB