Fill all the columns >= K
for i in range ( 0 , N ) : NEW_LINE INDENT for j in range ( K - 1 , N ) : NEW_LINE INDENT element += 1 ; NEW_LINE mat [ i ] [ j ] = element ; NEW_LINE DEDENT DEDENT return mat ; NEW_LINE