Was attempting a question in codechef and got TLE, so can anyone help me optimize it?

Question

Question: You are given two positive integers N and K, where K≤N. Find a sequence A1,A2,…,AN such that: for each valid i, Ai is either i or −i there are exactly K values of