[python] 특별한 이차원 배열 1

문제 설명

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

입출력 예

제출 내역

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

다른 풀이

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

프로그래머스 코딩 기초 트레이닝 Day24 조건문 활용, 반복문 활용, 이차원 리스트(배열)

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

💾