(연습문제) list 원소 추가 및 요소 검사하기

A형 

list 크기를 키보드로 입력 받은 후, 입력 받은 크기만큼 임의 숫자를 list에 추가하고, list의 크기를 출력하시오.

<출력 결과 예시>

vector 수 : 3

4

2

5

vector 크기 : 3

import random
lst = []
size = int(input("vector 수 : "))
for i in range(size):
    r = random.randint(0, size)
    lst.append(r)

for i in range(size):
    print(lst[i])
print("vector 크기 :", size)

- '임의의 숫자'를 입력받기 위해 random 모듈을 import 했다. (line 1)

- 목록을 받을 빈 리스트를 작성했다. (line 2)

- for 조건 (입력받은 수 만큼의 범위에서, 랜덤 수를 추출하고 조건에 맞으면 list에 추가한다.)

- list의 크기 (입력받은 수 = 범위 = list 내 개체 개수) 를 출력했다. 

 

 

B형

list 크기를 키보드로 입력 받은 후, 입력 받은 크기만큼 임의 숫자를 list에 추가한다. 이후 list에서 찾을 값을 키보드로 입력한 후 해당 값이 list에 있으면 "YES", 없으면 "NO"를 출력하시오.

<출력 결과 예시>

vector 수 : 5

1

2

3

4

5

3

YES

import random
lst = []
size = int(input("vector 수 : "))
for i in range(size):
    r = random.randint(0, size)
    lst.append(r)

for i in range(size):
    print(lst[i])

if int(input()) in lst:
    print('YES')
else:
    print('NO')

- 위와 같은 풀이에서 요소를 검사하기 위한 if 문이 추가되었다.

- 마지막 if문 조건에서, lst를 비교하는 것이므로 == 는 성립하지 않는다.