백준 1978번 소수찾기
·
알고리즘/백준
https://www.acmicpc.net/problem/1978 1978번: 소수 찾기첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.www.acmicpc.net문제주어진 수 N개 중에서 소수가 몇 개인지 찾아서 출력하는 프로그램을 작성하시오.입력 첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.출력주어진 수들 중 소수의 개수를 출력한다.예제 입력 1 복사41 3 5 7예제 출력 1 복사3문제풀이숫자를 입력받을 때 map함수를 사용하여 int 타입과 공백을 구분하기 위해 split함수를 사용split을 적용할 경우 예 1 12 3 출력 1 , 12 , 3..
[파이썬 알고리즘 인터뷰] 배열 - 세수의 합
·
알고리즘
문제설명 - 배열을 입력받아 함으로 0을 만들 수 있는 3개의 엘리먼트를 출력하라. 예시 #입력 nums = [-1, 0, 1, 2, -1, -4] #출력 [ [-1, 0, 1], 【-1, -1, 2] ] 풀이 앞 뒤로 같은 값이 있을 경우를 대비해 sort() 함수를 사용 중복된 값이 있을수 도 있기 때문에 continue로 건너 뛴다. 세 개의 포인터를 이용하여 각각의 포인터가 이동하면서 i + j + k = 0을 찾아내서 반환을 해준다. def three_sum(nums): results = [] nums.sort() print(nums) # 앞 뒤로 같은 값이 있을 경우를 대비해 sort() 함수를 사용 for i in range(len(nums) - 2): if i > 0 and nums[i] ..
[파이썬 알고리즘 인터뷰] 배열 - 빗물 트래핑
·
알고리즘
문제설명 - 높이를 입력받아 비 온 후 얼마나 많은 물이 쌓일 수 있는지 계산하라. 예시 #입력 [0,1,0,2,1,0,1,3,2,1,2,1] #출력 6 풀이 left는 height 배열 왼쪽부터 시작을 하고 right는 배열 오른쪽부터 시작을 한다. left_max 와 right_max는 height 배열의 원소값이다. 둘의 값을 비교하여서 left값이 더 작거나 같을 때는 left 배열값을 1 더해주어 점점 오른쪽으로 이동을하고 반대로 left값이 더 클경우 right 배열값을 1감소시켜 왼쪽으로 이동 시킨다. 비교를 하면서 좌우 기둥 최대높이 Left_max, right_max가 현재높이와의 차이만큼 물높이 volume 을더해나간다. def trap(height): volume = 0 left, ..
[파이썬 알고리즘 인터뷰] 배열 - 두 수의 합
·
알고리즘
문제 설명 - 덧셈하여 타겟을 만들 수 있는 배열의 두 숫자 인덱스를 리턴하라. 예시 #입력 nums = (2, 7, 11, 15], target = 9 #출력 [0, 1] #설명 nums[0] + nums[1]= 2 + 7 = 9 풀이 enumerate를 사용하여 리스트 원에 순서값을 부여해준다. 예시로 ( 0 , '2' ) 튜플형태로 담겨 있다. complement가 7이 되면 nums 배열안에 같은 값을 찾아보고 맞으면 리스트 형태로 반환한다. nums = [2, 7, 11, 15] # enumerate를 사용하여 리스트 원에 순서값을 부여해준다. # 예로 ( 0 , '2') 튜플형태로 담겨 있다. target = 9 def twoSum(nums: list, target: int) -> list: ..