mng giups minh vs, minh can gap
Cho dãy a gồm n phần tử có giá trị nguyên a1, a2, ..., an và một số nguyên k. Hãy cho biết số k có tồn tại trong dãy a hay không, nếu có thì in ra YES, ngược lại in NO và chèn vào vị trí [n/2] của dãy giá trị k, sau cùng là in lại dãy a. Biết [n/2] là giá trị nguyên của n khi chia cho 2.
Dữ liệu vào:
- Dòng đầu tiên là số nguyên dương n. (0 < n≤ 106) và số nguyên k (|k| ≤ 109).
- n dòng sau, dòng thứ i chứa một số nguyên ai (|ai| ≤ 109).
Kết quả: Ghi ra YES nếu số k tồn tại trong dãy a, ngược lại ghi ra 2 dòng như sau:
- Dòng 1: NO
- Dòng 2: Mảng a sau khi đã chèn theo yêu cầu. (Các số của mảng a được ghi trên 1 dòng, phân cách nhau bởi một dấu cách).
def check_and_insert(n, k, a):
if k in a:
return "YES"
else:
a.insert(n//2, k)
return "NO", a
n, k = map(int, input().split())
a = [int(input()) for _ in range(n)]
result = check_and_insert(n, k, a)
if result == "YES":
print(result)
else:
print(result[0])
print(' '.join(map(str, result[1])))