알고리즘/LeetCode
2148. Count Elements With Strictly Smaller and Greater Elements
앤테바
2022. 12. 27. 20:57
반응형
제목
문제)
2148. Count Elements With Strictly Smaller and Greater Elements
Given an integer array nums, return the number of elements that have both a strictly smaller and a strictly greater element appear in nums.
Example 1:
Input: nums = [11,7,2,15]
Output: 2
Explanation: The element 7 has the element 2 strictly smaller than it and the element 11 strictly greater than it.
Element 11 has element 7 strictly smaller than it and element 15 strictly greater than it.
In total there are 2 elements having both a strictly smaller and a strictly greater element appear in nums.
Example 2:
Input: nums = [-3,3,3,90]
Output: 2
Explanation: The element 3 has the element -3 strictly smaller than it and the element 90 strictly greater than it.
Since there are two elements with the value 3, in total there are 2 elements having both a strictly smaller and a strictly greater element appear in nums.
Constraints:
- 1 <= nums.length <= 100
- -105 <= nums[i] <= 105
솔루션1)
- 오름차순 정렬 후 배열을 순회하면서 해당 element가 양 끝단 값 사이에 있는지 체크
class Solution:
def countElements(self, nums: List[int]) -> int:
if len(nums) < 3:
return 0
nums.sort()
count = 0
for i in range(1, len(nums)-1):
if nums[0] < nums[i] < nums[-1]:
count += 1
return count
솔루션2)
- 솔루션1을 정렬없이 코드 수정
class Solution:
def countElements(self, nums: List[int]) -> int:
min_val = min(nums)
max_val = max(nums)
return sum([1 for n in nums if min_val < n < max_val])
반응형