LeetCode --- 412周赛

题目列表

3264. K 次乘运算后的最终数组 I

3266. K 次乘运算后的最终数组 II

3265. 统计近似相等数对 I

3267. 统计近似相等数对 II

一、K次乘预算后的最终数组 I & II

I 数据范围比较小,可以暴力模拟,代码如下

cpp 复制代码
class Solution {
public:
    vector<int> getFinalState(vector<int>& nums, int k, int multiplier) {
        int n = nums.size();
        while(k--){
            int pos = 0;
            for(int i = 1; i < n; i++){
                if(nums[pos] > nums[i])
                    pos = i;
            }
            nums[pos] *= multiplier;
        }
        return nums;
    }
};

II 的数据范围变大了,暴力会超时,需要优化,如何做?

这里我们需要找到一些规律:

1、假设数组nums中只有一个数字x,我们直接返回 x * multiplier ^ k

2、假设数组nums中有两个数字 x,y (x <= y)

  1. x 太小(或者 k 太小 或者 multiplier 太小),导致 x * multiplier ^ k < y

  2. x 和 y 相近,x 乘以 m (m < k) 个的 multiplier 后,就会大于 y,即 x * multiplier ^ m > y,那么下一次 y * multiplier 一定大于 x * multiplier ^ m (我们对两边同除multiplier,可得y > x*multiplier^(m-1) 成立),所以后面的乘法操作就是这两个数字交替进行。

3、假设数组nums中有三个数字 x,y,z (x <= y <= z),当 z 成为最小值,x 变成 X,y变成Y 时,假设 z <= Y <= X,由 2 可知,z * multiplier 必然大于 Y 和 X,下一个 Y * multiplier 必然大于 z,下一个 X * multiplier 必然大于 Y,同理后面的乘法操作就是这三个数字交替进行。

......

我们不难得出结论:当数组中的最大值成为最小值的时候,剩余的 操作次数 会被均匀的分配给每个数字,如果除不尽的化,会按照大小顺序分配多余的1次乘法操作

那么在数组中的最大值成为最小值之前的操作,我们只需要模拟即可,代码如下

cpp 复制代码
class Solution {
    const int MOD = 1e9 + 7;
    using LL = long long;
public:
    LL POW(LL x, LL y){
        LL res = 1;
        while(y){
            if(y & 1) res = (res * x) % MOD;
            x = (x * x) % MOD;
            y >>= 1; 
        }
        return res;
    }
    vector<int> getFinalState(vector<int>& nums, int k, int multiplier) {
        if(multiplier == 1) return nums;
        int n = nums.size();
        int mx = ranges::max(nums);
        priority_queue<pair<LL,int>,vector<pair<LL,int>>,greater<>>pq;
        // 建堆
        for(int i = 0; i < n; i++)
            pq.emplace(nums[i], i);
        // while循环的时间复杂度,考虑最坏情况 k = 1e9 ,muiliplier = 2
        // nums = {1,1,1,1,...,1e9}, U = max(nums), n = len(nums)
        // 即每一个 1 都需要 logU 的时间变得 >= 1e9
        // 时间复杂度为 min(k, nlognlogU) = nlognlogU
        while(k && mx != pq.top().first){
            auto [x, i] = pq.top(); pq.pop();
            pq.emplace(x * multiplier, i);
            k--;
        }
        int j = 0;
        int p1 = POW(multiplier, k/n) % MOD;
        int p2 = POW(multiplier, k/n + 1) % MOD;
        while(pq.size()){
            auto [x, i] = pq.top(); pq.pop();
            nums[i] = x % MOD * (j < k % n ? p2 : p1) % MOD;
            j++;
        }
        return nums;
    }
};

二、统计近似相等数对 I

题目一数据范围比较小,可以直接暴力模拟,代码如下

cpp 复制代码
class Solution {
public:
    int countPairs(vector<int>& nums) {
        int n = nums.size(), ans = 0;
        auto check = [&](int i, int j)->bool{
            int x = nums[i], y = nums[j], ret = 0;
            int cnt1[10]{}, cnt2[10]{};
            while(x || y){
                if(x % 10 != y % 10){
                    cnt1[x % 10]++;
                    cnt2[y % 10]++;
                    ret++;
                }
                x /= 10, y /= 10;
            }
            for(int i = 0; i < 10; i++)
                if(cnt1[i] != cnt2[i])
                    return false;
            return ret == 0 || ret == 2;
        };
        for(int i = 0; i < n; i++){
            for(int j = i + 1; j < n; j++){
                ans += check(i, j);
            }
        }
        return ans;
    }
};

三、统计近似相等数对 II

这种找匹配个数的题,很容易想到两数之和,用哈希表,边记录数字出现次数,边统计结果,这题也是同理,我们边记录数字出现次数,边统计当前遍历到的数字能变成哪些数字,在将这些数字的出现次数加入答案即可,代码如下

cpp 复制代码
class Solution {
public:
    int countPairs(vector<int>& nums) {
        ranges::sort(nums);
        int ans = 0;
        unordered_map<int,int>mp;
        for(int x: nums){
            unordered_set<int> st = {x}; // 记录 x 能变成哪些数字
            string s = to_string(x);
            int m = s.size();
            for(int i = 0; i < m; i++){
                for(int j = i + 1; j < m; j++){
                    swap(s[i], s[j]);
                    st.insert(stoi(s));
                    for(int p = i + 1; p < m; p++){
                        for(int q = p + 1; q < m; q++){
                            swap(s[p], s[q]);
                            st.insert(stoi(s));
                            swap(s[p], s[q]); // 注意要恢复现场
                        }
                    }
                    swap(s[i], s[j]); // 注意要恢复现场
                }
            }
            for(auto e: st)
                ans += (mp.contains(e) ? mp[e] : 0);
            mp[x]++;
        }
        return ans;
    }
};
相关推荐
sp_fyf_202429 分钟前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-11-01
人工智能·深度学习·神经网络·算法·机器学习·语言模型·数据挖掘
香菜大丸1 小时前
链表的归并排序
数据结构·算法·链表
jrrz08281 小时前
LeetCode 热题100(七)【链表】(1)
数据结构·c++·算法·leetcode·链表
oliveira-time1 小时前
golang学习2
算法
面试鸭1 小时前
离谱!买个人信息买到网安公司头上???
java·开发语言·职场和发展
南宫生2 小时前
贪心算法习题其四【力扣】【算法学习day.21】
学习·算法·leetcode·链表·贪心算法
懒惰才能让科技进步3 小时前
从零学习大模型(十二)-----基于梯度的重要性剪枝(Gradient-based Pruning)
人工智能·深度学习·学习·算法·chatgpt·transformer·剪枝
Ni-Guvara3 小时前
函数对象笔记
c++·算法
测试19983 小时前
2024软件测试面试热点问题
自动化测试·软件测试·python·测试工具·面试·职场和发展·压力测试
泉崎3 小时前
11.7比赛总结
数据结构·算法