본문 바로가기
Python/알고리즘

[COS PRO 2급 #3-10] 내 절반이 여기에 있는가?

by 포푸리 (POPOOLY) 2023. 1. 20.
반응형

문제설명

리스트 원소인 자신을 2로 나눈 값이 리스트에 있는 수의 개수를 구하려고 합니다. 예를 들어, 리스트가 [4, 8, 3, 6, 7]인 경우, 6/2 = 3, 8/2 =4이므로 자신을 2로 나눈 값이 리스트에 들어있는 수의 개수는 총 2개입니다. 
숫자가 담긴 리스트 arr가 주어졌을 때, 자신을 2로 나눈 값이 리스트에 들어있는 수가 몇 개인지 return 하는 solution 함수를 작성했습니다. 그러나, 코드 일부분이 잘못되어있기 때문에, 코드가 올바르게 동작하지 않습니다. 주어진 코드에서 _**한 줄**_만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정해주세요. 

매개변수 설명

숫자가 담긴 리스트 arr가 solution 함수의 매개변수로 주어집니다. 

* arr의 크기는 1 이상 1,000 이하입니다. 
* arr의 원소는 1,000 이하의 자연수입니다.

return 값 설명

자신을 2로 나눈 값이 리스트에 들어있는 수가 몇 개인지 return 해주세요. 

예시

| arr |return | 
|:---|:---:|---|:---:|---|:---:|--|:--------:|--------:| 
| [4, 8, 3, 6, 7] |2| 

예시설명

4/2=2이고, 2는 리스트에 없습니다. 
8/2=4이고 4는 4 이고 4는 리스트에 존재합니다. 
3/2=1.5이고 1.5는 리스트에 없습니다. 
6/2=3이고 3은 리스트에 존재합니다. 
7/2=3.5이고 3.5는 리스트에 없습니다. 
따라서 자신을 2로 나눈 값이 리스트에 들어있는 수가 총 2개이므로 2를 return 합니다.

문제 코드

def solution(arr):
    answer = 0
    for i in arr:
        for i/2 in arr:
            answer += 1
    return answer

풀이

def solution(arr):
    answer = 0
    for i in arr:
    	# 자신을 2로 나눈 수가 arr에 있으면 answer += 1
        if i/2 in arr:
            answer += 1
    return answer

댓글