【算法题】 8038. 收集元素的最少操作次数
2023-09-30 本文已影响0人
程序员小2
题目:
给你一个正整数数组 nums 和一个整数 k 。
一次操作中,你可以将数组的最后一个元素删除,将该元素添加到一个集合中。
请你返回收集元素 1, 2, ..., k 需要的 最少操作次数 。
示例 1:
输入:nums = [3,1,5,4,2], k = 2
输出:4
解释:4 次操作后,集合中的元素依次添加了 2 ,4 ,5 和 1 。此时集合中包含元素 1 和 2 ,所以答案为 4 。
示例 2:
输入:nums = [3,1,5,4,2], k = 5
输出:5
解释:5 次操作后,集合中的元素依次添加了 2 ,4 ,5 ,1 和 3 。此时集合中包含元素 1 到 5 ,所以答案为 5 。
示例 3:
输入:nums = [3,2,5,3,1], k = 3
输出:4
解释:4 次操作后,集合中的元素依次添加了 1 ,3 ,5 和 2 。此时集合中包含元素 1 到 3 ,所以答案为 4 。
提示:
1 <= nums.length <= 50
1 <= nums[i] <= nums.length
1 <= k <= nums.length
输入保证你可以收集到元素 1, 2, ..., k 。
java代码:
class Solution {
public int minOperations(List<Integer> nums, int k) {
int n = nums.size();
if(n==k) {
return n;
}
int count =0;
Set<Integer> set = new HashSet<>();
for(int i=n-1;i>=0;i--) {
count++;
if(nums.get(i)<=k) {
set.add(nums.get(i));
}
if(set.size()==k) {
break;
}
}
return count;
}
}