Leetcode Java Number of Subsequences That Satisfy the Given Sum Condition
업데이트:
문제
코드
class Solution {
private static final int MOD = 1000000007;
public int numSubseq(int[] nums, int target) {
Arrays.sort(nums);
int result = 0;
int length = nums.length;
int[] pows = new int[length];
pows[0] = 1;
for (int i = 1; i < length; i++) {
pows[i] = (2 * pows[i - 1]) % MOD;
}
int left = 0;
int right = length - 1;
while (left <= right) {
if (nums[left] + nums[right] > target) {
right--;
} else {
result = (result + pows[right - left++]) % MOD;
}
}
return result;
}
}
결과
설명
- 정수 배열 nums를 이용하여 부분 수열의 값이 최댓값과 최솟값의 차잇값이 target 이하인 부분 수열의 갯수를 계산하는 문제이다.
- 단, 값이 매우 클 수 있으므로, 모듈러 $10^9 + 7$을 적용한다.
- 문제 풀이에 필요한 변수를 정의 및 정렬한다.
- nums를 오름차순으로 정렬한다.
- result는 부분 수열의 갯수를 계산할 변수로, 0으로 초기화한다.
- length는 nums의 길이를 저장한 변수이다.
- pows는 2의 거듭제곱값을 저장할 변수로, length 길이의 정수 배열로 초기화하고 첫 값을 1로 넣어주고 각 위치에 해당 위치에 해당하는 2의 거듭제곱 값을 모듈러를 적용한 값으로 넣어준다.
- left와 right는 nums의 위치 변수로, 0과 $length - 1$로 초기화한다.
- left가 right 이하까지 아래를 반복한다.
- $nums[left] + nums[right]$의 값이 target보다 큰 경우, right를 감소시켜 범위를 좁혀준다.
- 위의 경우가 아니라면 조건에 만족하므로, result에 $result + pows[right - left]$ 값에 모듈러를 적용한 값으로 넣어준다.
- 반복이 완료되면 계산된 부분 수열의 갯수인 result를 주어진 문제의 결과로 반환한다.
소스
Sample Code는 여기에서 확인 가능합니다.
댓글남기기