[프로그래머스/파이썬] Lv.0 특별한 이차원 배열 1 (10')

Programmers 레벨 0_코딩 기초 트레이닝 문제 풀이

https://school.programmers.co.kr/learn/challenges/training?order=recent

코딩 기초 트레이닝 | 프로그래머스 스쿨

 

 

풀이 날짜 : 2023-09-21

소요 시간 : 10min

 

---

 

문제 설명

정수 n이 매개변수로 주어질 때, 다음과 같은 n × n 크기의 이차원 배열 arr를 return 하는 solution 함수를 작성해 주세요.
arr[i][j] (0 ≤ i, j < n)의 값은 i = j라면 1, 아니라면 0입니다.

 

 

 

나의 풀이

def solution(n):
    answer = [[0 for j in range(n)] for i in range(n)]
    # answer = [[0]*n for i in range(n)]
    
    for a in range(n) :
        answer[a][a] += 1
    
    return answer

 

 

 

다른 풀이

def solution(n):
    return [[1 if i==j else 0 for j in range(n)] for i in range(n)]