LumberUolevi

input/code.cpp: In function 'int main()':
input/code.cpp:74:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   74 |         for (int i = 0; i < vals.size() - 1; i++)
      |                         ~~^~~~~~~~~~~~~~~~~

#include <iostream>
#include <vector>
#include <algorithm>
 
int main()
{
    std::vector<int> vals = std::vector<int>();
    
	int n, k;
	std::cin >> n >> k;
    int total = n;
	if (k >= n)
	{
		for (int i = 1; i < n; i++)
			std::cout << i << " ";
		std::cout << n;
		return 0;
	}
 
	if (k == 0)
		return 0;
 
    if (k > 3)
    {
	    vals.push_back(n);
	    n--;
	    k--;
    }
    else if (k == 2)
    {
        std::cout << 1 << " " << n - 1;
        return 0;
    }
	
	if (k == 0)
	    return 0;
 
	int i = 1;
	
	while (k > 3)
	{
	    vals.push_back(i);
	    vals.push_back(n);
 
		i++;
		n--;
		k -= 2;
	}
	if (k == 1)
		vals.push_back(n);
	else if (k == 2)
	{
	    vals.push_back(i);
	    vals.push_back(n);
	}
	else
	{
	    // Now we need the 3 last ones
	    if (total - (i + i + 1) <= n && total - (i + i + 1) > i + 1)
	    {
	        vals.push_back(i);
	        vals.push_back(i + 1);
	        vals.push_back(total - (i + i + 1));
	    }
	    else
	    {
	        vals.push_back(i);
	        vals.push_back(n);
	        vals.push_back(n - 1);
	    }
	}
	
	std::sort(vals.begin(), vals.end());
	for (int i = 0; i < vals.size() - 1; i++)
	    std::cout << vals[i] << " ";
    std::cout << vals[vals.size() - 1];
	
	return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: WRONG ANSWER

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED

Test 22

Verdict: ACCEPTED

Test 23

Verdict: ACCEPTED

Test 24

Verdict: ACCEPTED

Test 25

Verdict: ACCEPTED

Test 26

Verdict: ACCEPTED

Test 27

Verdict: ACCEPTED

Test 28

Verdict: ACCEPTED

Test 29

Verdict: WRONG ANSWER

Test 30

Verdict: ACCEPTED

Test 31

Verdict: ACCEPTED

Test 32

Verdict: ACCEPTED

Test 33

Verdict: ACCEPTED

Test 34

Verdict: ACCEPTED

Test 35

Verdict: WRONG ANSWER

Test 36

Verdict: ACCEPTED

Test 37

Verdict: WRONG ANSWER

Test 38

Verdict: ACCEPTED

Test 39

Verdict: ACCEPTED

Test 40

Verdict: ACCEPTED

Test 41

Verdict: ACCEPTED

Test 42

Verdict: ACCEPTED

Test 43

Verdict: ACCEPTED

Test 44

Verdict: ACCEPTED

Test 45

Verdict: ACCEPTED

Test 46

Verdict: ACCEPTED

Test 47

Verdict: ACCEPTED

Test 48

Verdict: ACCEPTED

Test 49

Verdict: ACCEPTED

Test 50

Verdict: ACCEPTED

Test 51

Verdict: ACCEPTED

Test 52

Verdict: ACCEPTED

Test 53

Verdict: ACCEPTED

Test 54

Verdict: ACCEPTED

Test 55

Verdict: WRONG ANSWER

Test 56

Verdict: ACCEPTED

Test 57

Verdict: ACCEPTED

Test 58

Verdict: ACCEPTED

Test 59

Verdict: WRONG ANSWER

Test 60

Verdict: ACCEPTED

Test 61

Verdict: ACCEPTED

Test 62

Verdict: ACCEPTED

Test 63

Verdict: ACCEPTED