[HackerRank] Grid Challenge

Problem

Code

#!/bin/python3

import math
import os
import random
import re
import sys

#
# Complete the 'gridChallenge' function below.
#
# The function is expected to return a STRING.
# The function accepts STRING_ARRAY grid as parameter.
#

def gridChallenge(grid):
    # Write your code here
    row_len = len(grid[0])
    col_len = len(grid)
    
    '''
    Step 1.
    grid 내에 알파벳 소문자 문자열들을 정렬된 상태로 list로 변환 후,
    n_grid에 append
    '''
    n_grid = list()
    for row in grid:
        row = sorted(list(row))
        n_grid.append(row)
    
    '''
    Step 2.
    각 column의 알파벳들이 알파벳 순서대로 정렬되어 있는지 검사
    '''
    for i in range(row_len):
        tmp = []
        for j in range(col_len):
            tmp.append(n_grid[j][i])
        tmp2 = sorted(tmp)
        if tmp != tmp2:
            return 'NO'
    return 'YES'

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    t = int(input().strip())

    for t_itr in range(t):
        n = int(input().strip())

        grid = []

        for _ in range(n):
            grid_item = input()
            grid.append(grid_item)

        result = gridChallenge(grid)

        fptr.write(result + '\n')

    fptr.close()

Reference

HackerRank - Grid Challenge