codechef/POSPREFS.cpp at main · Akanksha2403/codechef · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#include <iostream>
using namespace std;
int main()
{
    int t,j;
    cin>>t;
    while(t--)
    {
     int n,k,i,sum=0;
     cin>>n>>k;

     int a[n+1];
     for(i=1;i<=n;)
     {

      if(k==0||k==n)
       break;
      if(sum+i<=i+1 && k>0)
      {
       a[i]=i;
       sum+=i;
       i++;
       k--;
       continue;
   }
   if(sum>i)
   {
    a[i]=-i;
    sum-=i;
    i++;
    if(sum>0)
     k--;
    continue;
   }
   if(sum+i >i+1 &&k==1)
   {
    a[i]=-i;
    i++;
    if(sum-i>0)
     break;
    else
    {
     sum-=i;
     continue;
    }
   }

   if(sum+i >i+1 &&k>1)
   {
    a[i]=i;
    if(sum>0)
     k--;
    sum+=i;
    i++;
   }

  }
  if(k==n)
  {
   for(i=1;i<=n;i++)
    a[i]=i;
  }

  else if(i<=n)
  {
   for(;i<=n;i++)
    a[i]=-i;
  }

  for(i=1;i<=n;i++)
   cout<<a[i]<<" ";

  cout<<endl;
 }
    return 0;
}