Ma trận vuông A cấp n được gọi là ma trận băng với bán kính băng k (band matrix) nếu Aij=0 với mọi i,j thoả ∣i−j∣>k (các chỉ số hàng, cột đánh số từ 1). Cho ma trận A và số k, hãy kiểm tra A có phải ma trận băng với bán kính k hay không.
Dòng đầu chứa hai số nguyên n,k (0≤k≤n−1). n dòng tiếp theo, mỗi dòng gồm n số nguyên, là các phần tử của A.
1≤n≤30; 0≤k≤n−1; phần tử có giá trị tuyệt đối không vượt quá 1000.
In ra "YES" nếu A là ma trận băng bán kính k, ngược lại in "NO".
Ví dụ:
Đầu vào:
1 0
5
Đầu ra:
YES
Đang tải editor...