力扣第284场周赛题解

A.6031. 找出数组中的所有 K 近邻下标

找到每个$nums[j]==key$的$j$，将所有$max(0,j-k),min(j+k,n-1)$的下标都统计起来即可。$1<=nums.length<=1000$，不需要优化可以直接$O(n^2)$过。

AC代码：

class Solution
{
public:
    vector<int> findKDistantIndices(vector<int> &nums, int key, int k)
    {
        int n = nums.size();
        set<int> st;
        for (int i = 0; i < n; i++)
            if (nums[i] == key)
                for (int j = max(0, i - k); j <= min(n - 1, i + k); j++)
                    st.insert(j);
        vector<int> ans;
        for (auto it : st)
            ans.emplace_back(it);
        return ans;
    }
};

B.5203. 统计可以提取的工件

看上去题目很复杂，但是题目保证了每个$dig$元素都不重复，也即每次挖掘都不重复，那么只需要统计面积 == 挖掘次数的元件数量即可。

AC代码：

class Solution
{
public:
    int digArtifacts(int n, vector<vector<int>> &artifacts, vector<vector<int>> &dig)
    {
        map<int, int> num;
        int mp[1010][1010];
        memset(mp, 0, sizeof mp);
        int id = 0;//给元件编号
        for (auto it : artifacts)
        {
            num[++id] = (it[2] - it[0] + 1) * (it[3] - it[1] + 1);
            for (int i = it[0]; i <= it[2]; i++)
                for (int j = it[1]; j <= it[3]; j++)
                    mp[i][j] = id;
        }
        int ans = 0;
        for (auto it : dig)
        {
            int val = mp[it[0]][it[1]];
            if (num[val] == 1)
                ans++;
            else
                num[val]--;
        }
        return ans;
    }
};

C.5227. K 次操作后最大化顶端元素

前排也wa的挺惨的，个人感觉题目有歧义。

两种情况：

1. 先使用$k-1$次把前$k-1$个元素删除，最后一次恢复已删除中的最大值。
2. 删除前$k$个元素。

还有 $-1$ 以及 $k>n$的特殊情况，special judge 即可。

AC代码：

class Solution
{
public:
    int maximumTop(vector<int> &nums, int k)
    {
        int ans = 0, mx = 0;
        int n = nums.size();
        if (n == 1 && k % 2)
            return -1;
        for (auto it : nums)
            mx = max(mx, it);
        if (k > n)
            return mx;

        int p = min(n, k - 1);
        for (int i = 0; i < p; i++)
            ans = max(ans, nums[i]);
        if (p + 1 < n)
            ans = max(ans, nums[p + 1]);
        return ans;
    }
};

D.6032. 得到要求路径的最小带权子图

1. 从起点1出发，求出到各个点的最短距离$dis1[]$
2. 从起点2出发，求出到各个点的最短距离$dis2[]$
3. 建反图，从终点出发，求出到各个点的最短距离$dis3[]$

答案就是${\min }_0^{n-1}dis1[i]+dis2[i]+dis3[i]$

AC代码：

class Solution
{
public:
    long long minimumWeight(int n, vector<vector<int>> &edges, int src1, int src2, int dest)
    {
        const int N = 1e5 + 5;
        vector<pair<int, int>> mp[N], rev_mp[N];
        long long dis1[N], dis2[N], dis3[N];
        int vis1[N], vis2[N], vis3[N];

        for (auto it : edges)
        {
            mp[it[0]].emplace_back(make_pair(it[1], it[2]));
            rev_mp[it[1]].emplace_back(make_pair(it[0], it[2]));
        }
        for (int i = 0; i < n; i++)
        {
            dis1[i] = dis2[i] = dis3[i] = 1e18;
            vis1[i] = vis2[i] = vis3[i] = 0;
        }

        priority_queue<pair<long long, int>> q;
        dis1[src1] = 0;
        q.push(make_pair(0, src1));
        while (q.size()) // src1 出发 单源最短路
        {
            int u = q.top().second;
            long long w = -q.top().first;
            q.pop();
            if (vis1[u])
                continue;
            vis1[u] = 1;
            for (auto it : mp[u])
            {
                int v = it.first;
                if (w + it.second < dis1[v])
                {
                    dis1[v] = w + it.second;
                    q.push(make_pair(-dis1[v], v));
                }
            }
        }

        dis2[src2] = 0;
        q.push(make_pair(0, src2));
        while (q.size()) // src2 出发 单源最短路
        {
            int u = q.top().second;
            long long w = -q.top().first;
            q.pop();
            if (vis2[u])
                continue;
            vis2[u] = 1;
            for (auto it : mp[u])
            {
                int v = it.first;
                if (w + it.second < dis2[v])
                {
                    dis2[v] = w + it.second;
                    q.push(make_pair(-dis2[v], v));
                }
            }
        }

        dis3[dest] = 0;
        q.push(make_pair(0, dest));
        while (q.size()) //反图  dest 出发 单源最短路
        {
            int u = q.top().second;
            long long w = -q.top().first;
            q.pop();
            if (vis3[u])
                continue;
            vis3[u] = 1;
            for (auto it : rev_mp[u])
            {
                int v = it.first;
                if (w + it.second < dis3[v])
                {
                    dis3[v] = w + it.second;
                    q.push(make_pair(-dis3[v], v));
                }
            }
        }
        long long ans = 1e18;
        for (int i = 0; i < n; i++)
            ans = min(ans, dis1[i] + dis2[i] + dis3[i]);
        return ans == 1e18 ? -1 : ans;
    }
};