배열 에 몇 개의 그룹 과 k 보다 작은 세 개의 python 버 전이 있 는 지 찾 아 보 세 요.

802 단어 면접시험
n 길이 의 배열 Arr 와 하나의 수 k 를 제시 하여 배열 에 몇 개의 그룹 과 k 이하 의 세 개의 수 를 찾 습 니 다. 즉, Arr [a] + Arr [b] + arr [c] 입력: 6 (첫 번 째 행 위 는 하나의 정수 n, 입력 배열 의 길 이 를 대표 합 니 다) - 20, 1, 2, 3, 6 (두 번 째 행 위 는 하나의 정수 k) 출력: 4 (하나의 정수, 몇 개의 그룹 이 조건 을 만족 시 키 는 3 원 그룹 이 있 는 지 표시 합 니 다)
import sys
num = int(sys.stdin.readline().strip().split()[0])
arr = map(int, sys.stdin.readline().strip().split())
k = int(sys.stdin.readline().strip().split()[0])
sorted(arr)
print arr
count = 0
for index in range(num):
    i = index + 1
    j = num-1
    while i < j:
        if (arr[i] + arr[j] < k - arr[index]):
            # print i,j,index
            count+=1
            print i,j,index
            j-=1
        else:
            j -= 1
        if i==j:
            i+=1
            j=num-1
print count

좋은 웹페이지 즐겨찾기