LeetCode 340场周赛

LeetCode 340场周赛

Q1

暴力

class Solution {
   public:
    bool isPrime(int x) {
        if (x == 1) return false;
        for (int i = 2; i * i <= x; i++) {
            if (x % i == 0) return false;
        }
        return true;
    }
    int diagonalPrime(vector<vector<int>>& nums) {
        int n = nums.size(), m = nums[0].size(), ans = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (i == j && isPrime(nums[i][j])) {
                    ans = max(ans, nums[i][j]);
                }
                if (i + j == n - 1 && isPrime(nums[i][j])) {
                    ans = max(ans, nums[i][j]);
                }
            }
        }
        return ans;
    }
};

Q2

前缀和STL容器

#include 
using namespace std;
typedef long long ll;
const int N = 1e5 + 7;
const int mod = 1e9 + 7;

// 给你一个下标从 0 开始的整数数组 nums 。现有一个长度等于 nums.length 的数组 arr 。对于满足 nums[j] == nums[i] 且 j != i 的所有 j ，arr[i] 等于所有 |i - j| 之和。如果不存在这样的 j ，则令 arr[i] 等于 0 。

// 返回数组 arr 。

class Solution {
   public:
    vector<long long> distance(vector<int> &nums) {
        int n = nums.size();  // 1e5
        map<int, vector<long long>> mp;
        for (int i = 0; i < n; i++) {
            mp[nums[i]].push_back(i);
        }
        map<int, vector<long long>> copy = mp;
        for (auto &[k, v] : mp) {
            for (int i = 1; i < v.size(); i++) {
                v[i] += v[i - 1];
            }
        }

        vector<long long> ans(n);
        map<int, int> cnt;

        for (int i = 0; i < n; i++) {
            cnt[nums[i]]++;
            int pos = cnt[nums[i]] - 1;
            vector<long long> &a = mp[nums[i]];
            vector<long long> &b = copy[nums[i]];

            if (a.size() <= 1 || pos >= a.size() || pos<0) {
                ans[i] = 0;
                continue;
            }
            ll now = b[pos];
            int sz = a.size();
            ll presum = pos > 0 ? a[pos - 1] : 0, endsum = a[sz - 1] - a[pos];
            ans[i] = now * (pos) - presum + endsum - now * 1ll * (sz - pos - 1ll);
        }

        return ans;
    }
};

Q3

二分。

class Solution {
   public:
    int minimizeMax(vector<int>& nums, int p) {
        int n = nums.size();
        sort(nums.begin(), nums.end());
        vector<int> vis(n, 0);
        auto check = [&](int mid) {
            int cnt = 0, ok = 0;
            for (int i = 1; i < n; i++) {
                if (nums[i] - nums[i - 1] <= mid && !ok) {
                    // 前一个相邻的取了，后面一个就不能取了，相邻三个数取前后效果一样
                    cnt++;
                    ok = 1;
                } else {
                    ok = 0;
                }
            }
            return cnt >= p;
        };

        int l = 0, r = 1e9, ans = -1;
        while (l <= r) {
            int mid = l + r >> 1;
            if (check(mid)) {
                ans = mid;
                r = mid - 1;
            } else {
                l = mid + 1;
            }
        }
        return ans;
    }
};

只会三题，啊啊啊啊啊啊啊。

Q4

会暴力，但是不会优化。

感觉需要使用数据结构找到区间最小的那个。

没有时间写，写不来，不熟。

Gggggggg.