Editing 0511 - K Sort (section)

==Rezolvare==
<syntaxhighlight lang="python" line="1">
#0511 - K Sort
# definim functia care va citi si va returna un vector de n elemente, dat de la tastatura
def citeste_vector(n):
    return list(map(int, input().split()))
# definim functia care va returna primele k elemente in ordine crescatoare dintr-un vector dat
def ordonare_primele_k_elemente(k, vector):
    return sorted(vector[:k])
# definim functia care va returna ultimele n-k elemente in ordine descrescatoare
# dintr-un vector dat
def ordonare_ultimele_n_minus_k_elemente(n, k, vector):
    return sorted(vector[k:], reverse=True)

if __name__ == '__main__':
    n, k = map(int, input().split())
    vector = citeste_vector(n)
    # verificarea restrictiilor
    if n < 2 or n > 1000 or k < 1 or k >= n or any(x >= 1000000000 for x in vector):
        print("Datele de intrare nu corespund restricțiilor impuse.")
    else:
        print("Datele de intrare corespund restricțiilor impuse.")
        primele_k_elemente = ordonare_primele_k_elemente(k, vector)
        ultimele_n_minus_k_elemente = ordonare_ultimele_n_minus_k_elemente(n, k, vector)
        print(*(primele_k_elemente + ultimele_n_minus_k_elemente)) # afisam rezultatul 

   
   
</syntaxhighlight>