【LeetCode周赛】第 387 场周赛

目录

  • [100243. 将元素分配到两个数组中 I 简单](#100243. 将元素分配到两个数组中 I 简单)
  • [100237. 元素和小于等于 k 的子矩阵的数目 中等](#100237. 元素和小于等于 k 的子矩阵的数目 中等)
  • [100234. 在矩阵上写出字母 Y 所需的最少操作次数 中等](#100234. 在矩阵上写出字母 Y 所需的最少操作次数 中等)
  • [100246. 将元素分配到两个数组中 II 困难](#100246. 将元素分配到两个数组中 II 困难)

100243. 将元素分配到两个数组中 I 简单

100243. 将元素分配到两个数组中 I

分析:

根据题意模拟即可!

代码:

cpp 复制代码
class Solution {
public:
    vector<int> resultArray(vector<int>& nums) {
        vector<int> r1, r2, ans;
        r1.push_back(nums[0]),r2.push_back(nums[1]);
        int n=nums.size();
        for(int i=2;i<n;i++){
            int rr1 = r1[r1.size()-1], rr2 = r2[r2.size()-1];
            if(rr1>rr2) r1.push_back(nums[i]);
            else r2.push_back(nums[i]);
        }
        for(int i=0;i<r1.size();i++) ans.push_back(r1[i]);
        for(int i=0;i<r2.size();i++) ans.push_back(r2[i]);
        return ans;
    }
};

100237. 元素和小于等于 k 的子矩阵的数目 中等

100237. 元素和小于等于 k 的子矩阵的数目

分析:

二维前缀和,子矩阵必须包含原矩阵的最左上角的数,因此直接枚举所有矩阵的前缀和即可!

代码:

cpp 复制代码
class Solution {
public:
    int countSubmatrices(vector<vector<int>>& grid, int k) {
        if(grid[0][0]>k) return 0;
        int n=grid.size(),m=grid[0].size(),ans=0;
        vector<vector<int>> ma(n,vector<int>(m,0));
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(j==0) ma[i][j]=grid[i][j];
                else  ma[i][j]=grid[i][j]+ma[i][j-1];
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(i!=0) ma[i][j]+=ma[i-1][j];
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(ma[i][j]<=k) ans++;
            }
        }
        return ans;
    }
};

更为精简的写法:

cpp 复制代码
class Solution {
public:
    int countSubmatrices(vector<vector<int>>& grid, int k) {
        if(grid[0][0]>k) return 0;
        int n=grid.size(),m=grid[0].size(),ans=0;
        vector<vector<int>> ma(n+1,vector<int>(m+1,0));
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                ma[i+1][j+1] = ma[i+1][j] + ma[i][j+1] -ma[i][j] + grid[i][j];
                ans+= ma[i+1][j+1]<=k;
            }
        }
        return ans;
    }
};

100234. 在矩阵上写出字母 Y 所需的最少操作次数 中等

100234. 在矩阵上写出字母 Y 所需的最少操作次数

分析:

记录 属于Y不属于Y 的0、1、2的个数,再枚举各种可能,计算最小的操作数。

代码:

cpp 复制代码
class Solution {
public:
    int minimumOperationsToWriteY(vector<vector<int>>& grid) {
        int n=grid.size(),m=n/2;
        vector<int> y(3,0),o(3,0);
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if((i==j&&i<=m)||(i+j==n-1&&i<=m)||(j==m&&i>m)){
                    y[grid[i][j]]++;
                }else{
                    o[grid[i][j]]++;
                }
            }
        }
        int yy=y[0]+y[1]+y[2],oo=o[0]+o[1]+o[2],ans=n*n;
        for(int i=0;i<3;i++){
            for(int j=0;j<3;j++){
                if(i==j) continue;
                ans=min(ans,yy-y[i]+oo-o[j]);
            }
        }
        return ans;
    }
};

100246. 将元素分配到两个数组中 II 困难

100246. 将元素分配到两个数组中 II

分析:

我的方法(错误):

对于 arr1,再创建一个数组t1用于维持arr1中的数组有序,arr2同理。
nums[i]需要插入时,使用二分查找分别计算 arr1arr2大于 它的值的数量,同时计算插入t1或者t2的对应位置。

但是 vector.insert()的时间复杂度是 O(n),因此总体时间复杂度变为了O(n^2),会超时,但是提交之后也能过,但不是最优解。

离散化+树状数组:
题解

代码:

我的代码(错误):

cpp 复制代码
class Solution {
public:
    vector<int> resultArray(vector<int>& nums) {
        vector<int> r1,rr1,r2,rr2,ans;
        r1.push_back(nums[0]);r2.push_back(nums[1]);
        rr1.push_back(nums[0]);rr2.push_back(nums[1]);
        int n=nums.size();
        for(int i=2;i<n;i++){
            int l1=r1.size(), l2=r2.size();
            int i1=upper_bound(r1.begin(),r1.end(),nums[i]) - r1.begin(), i2=upper_bound(r2.begin(),r2.end(),nums[i]) - r2.begin();
            if(l1-i1>l2-i2){
                r1.insert(r1.begin()+i1,nums[i]);
                rr1.push_back(nums[i]);
            }else if(l1-i1<l2-i2){
                r2.insert(r2.begin()+i2,nums[i]);
                rr2.push_back(nums[i]);
            }else{
                if(l1<=l2){
                    r1.insert(r1.begin()+i1,nums[i]);
                    rr1.push_back(nums[i]);
                }else if(l1>l2){
                    r2.insert(r2.begin()+i2,nums[i]);
                    rr2.push_back(nums[i]);
                }
            }
        }
        rr1.insert(rr1.end(),rr2.begin(),rr2.end());
        return rr1;
    }
};

离散化+树状数组:

cpp 复制代码
class TreeVector{
private:
    vector<int> tree;
public:
    TreeVector(int n): tree(n){}

    void add(int v){
        int i = v;
        while(i<tree.size()){
            tree[i]+=1;
            i += (i & -i);
        }
    }

    int pre(int v){
        int res = 0;
        while(v>0){
            res+=tree[v];
            v &= v-1;
        }
        return res;
    }
};


class Solution {
public:
    vector<int> resultArray(vector<int>& nums) {
        auto sorted = nums;
        ranges::sort(sorted);
        sorted.erase(unique(sorted.begin(), sorted.end()), sorted.end());
        int m=sorted.size(),n=nums.size();

        TreeVector t1(m+1), t2(m+1);
        vector<int> r1{nums[0]},r2{nums[1]};
        t1.add(ranges::lower_bound(sorted, nums[0])-sorted.begin()+1);
        t2.add(ranges::lower_bound(sorted, nums[1])-sorted.begin()+1);
        for(int i=2;i<n;i++){
            int x=nums[i];
            int v = ranges::lower_bound(sorted, x) - sorted.begin() + 1;
            int c1 = r1.size() - t1.pre(v), c2 = r2.size() - t2.pre(v);
            if(c1 > c2 || c1 == c2 && r1.size()<=r2.size()){
                r1.push_back(nums[i]);
                t1.add(v);
            }else{
                r2.push_back(nums[i]);
                t2.add(v);
            }
        }
        r1.insert(r1.end(), r2.begin(), r2.end());
        return r1;
    }
};
相关推荐
白榆maple13 分钟前
(蓝桥杯C/C++)——基础算法(下)
算法
JSU_曾是此间年少18 分钟前
数据结构——线性表与链表
数据结构·c++·算法
sjsjs1124 分钟前
【数据结构-合法括号字符串】【hard】【拼多多面试题】力扣32. 最长有效括号
数据结构·leetcode
此生只爱蛋1 小时前
【手撕排序2】快速排序
c语言·c++·算法·排序算法
咕咕吖2 小时前
对称二叉树(力扣101)
算法·leetcode·职场和发展
九圣残炎2 小时前
【从零开始的LeetCode-算法】1456. 定长子串中元音的最大数目
java·算法·leetcode
lulu_gh_yu2 小时前
数据结构之排序补充
c语言·开发语言·数据结构·c++·学习·算法·排序算法
丫头,冲鸭!!!3 小时前
B树(B-Tree)和B+树(B+ Tree)
笔记·算法
Re.不晚3 小时前
Java入门15——抽象类
java·开发语言·学习·算法·intellij-idea
为什么这亚子4 小时前
九、Go语言快速入门之map
运维·开发语言·后端·算法·云原生·golang·云计算