【算法题】 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;
    }
}
上一篇下一篇

猜你喜欢

热点阅读