Leetcode.2735 收集巧克力

题目链接

Leetcode.2735 收集巧克力 rating : 2043

给你一个长度为 n n n、下标从 0 0 0 开始的整数数组 n u m s nums nums n u m s [ i ] nums[i] nums[i] 表示收集位于下标 i i i 处的巧克力成本。每个巧克力都对应一个不同的类型,最初,位于下标 i i i 的巧克力就对应第 i i i 个类型。

在一步操作中,你可以用成本 k k k 执行下述行为:

  • 同时修改所有巧克力的类型,将巧克力的类型 i t h i_{th} ith 修改为类型 ( ( i + 1 )   m o d   n ) ((i + 1)\ mod\ n) ((i+1) mod n) t h _{th} th

假设你可以执行任意次操作,请返回收集所有类型巧克力所需的最小成本。

示例 1:

输入:nums = [20,1,15], x = 5
输出:13
解释:最开始,巧克力的类型分别是 [0,1,2] 。我们可以用成本 1 购买第 1 个类型的巧克力。
接着,我们用成本 5 执行一次操作,巧克力的类型变更为 [1,2,0] 。我们可以用成本 1 购买第 2 个类型的巧克力。
然后,我们用成本 5 执行一次操作,巧克力的类型变更为 [2,0,1] 。我们可以用成本 1 购买第 0 个类型的巧克力。
因此,收集所有类型的巧克力需要的总成本是 (1 + 5 + 1 + 5 + 1) = 13 。可以证明这是一种最优方案。

示例 2:

输入:nums = [1,2,3], x = 4
输出:6
解释:我们将会按最初的成本收集全部三个类型的巧克力,而不需执行任何操作。因此,收集所有类型的巧克力需要的总成本是 1 + 2 + 3 = 6 。

提示:
  • 1 ≤ n u m s . l e n g t h ≤ 1000 1 \leq nums.length \leq 1000 1nums.length1000
  • 1 ≤ n u m s [ i ] ≤ 1 0 9 1 \leq nums[i] \leq 10^9 1nums[i]109
  • 1 ≤ k ≤ 1 0 9 1 \leq k \leq 10^9 1k109

解法:枚举

我们最多可以移动 n − 1 n - 1 n1 次,因为移动 n n n 次就复原了。

  • 如果不移动的话,巧克力 i i i 会花费 n u m s [ i ] nums[i] nums[i] 成本;
  • 移动一次,巧克力 i i i 会花费 m i n ( n u m s [ i ] , n u m s [ i + 1 ] ) min(nums[i],nums[i+1]) min(nums[i],nums[i+1]) 成本;
  • 移动两次,巧克力 i i i 会花费 m i n ( n u m s [ i ] , n u m s [ i + 1 ] , n u m s [ i + 2 ] ) min(nums[i],nums[i+1],nums[i+2]) min(nums[i],nums[i+1],nums[i+2]) 成本;

我们直接用 s [ i ] s[i] s[i] 表示移动 i i i 次,收集所有类型巧克力的成本。

  • 我们直接使用两个指针 i i i j j j 枚举子数组的左右端点;
  • 在枚举的过程中维护子数组的最小值 x x x
  • 最后把贡献 x x x 加入到 s [ j − i ] s[j - i] s[ji]中,因为 x x x 就是移动了 j − i j - i ji 次的子数组 [ i ∼ j ] [i\sim j] [ij] 的最小值。

时间复杂度: O ( n 2 ) O(n^2) O(n2)

C++代码:

using LL = long long;

class Solution {
public:
    long long minCost(vector<int>& nums, int k) {
        int n = nums.size();
        vector<LL> s(n);
        for(int i = 0;i < n;i++) s[i] = i * 1LL * k;

        for(int i = 0;i < n;i++){
            int x = nums[i];
            for(int j = i;j < n + i;j++){
                x = min(x , nums[j % n]);
                s[j - i] += x;
            }
        }

        return *min_element(s.begin(),s.end());
    }
};

你可能感兴趣的:(Leetcode,枚举)