第 129 场 LeetCode 双周赛题解

A 构造相同颜色的正方形

枚举:枚举每个 3 × 3 3\times 3 3×3的矩阵,判断是否满足条件

cpp 复制代码
class Solution {
  public:
    bool canMakeSquare(vector<vector<char>>& grid) {
        for (int i = 0; i < 2; i++)
            for (int j = 0; j < 2; j++) {
                int c1 = 0, c2 = 0;
                for (int r = 0; r < 2; r++)
                    for (int c = 0; c < 2; c++)
                        if (grid[i + r][j + c] == 'B')
                            c1++;
                        else
                            c2++;
                if (max(c1, c2) >= 3)
                    return true;
            }
        return false;
    }
};

B 直角三角形

枚举:记录各行各列的 1 1 1 的数目,然后枚举每个直接三角形的直角所在的位置 g r i d [ i ] [ j ] grid[i][j] grid[i][j]

cpp 复制代码
class Solution {
  public:
    long long numberOfRightTriangles(vector<vector<int>>& grid) {
        int m = grid.size(), n = grid[0].size();
        vector<int> row(m), col(n);
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++) {
                row[i] += grid[i][j];
                col[j] += grid[i][j];
            }
        long long res = 0;
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
                if (grid[i][j])
                    res += 1LL * (row[i] - 1) * (col[j] - 1);
        return res;
    }
};

C 找出所有稳定的二进制数组 I

动态规划:设 p [ i ] [ j ] [ t a i l ] p[i][j][tail] p[i][j][tail] 为含有 i i i 个 0 0 0 和 j j j 个 1 1 1 且以 t a i l tail tail 为结尾的稳定二进制数组的个数,可以枚举其全为 t a i l tail tail 的后缀数组的可能长度来进行状态转移

cpp 复制代码
class Solution {
  public:
    using ll = long long;
    ll mod = 1e9 + 7;
    int numberOfStableArrays(int zero, int one, int limit) {
        ll p[zero + 1][one + 1][2]; 
        memset(p, 0, sizeof(p));
        for (int cz = 1; cz <= zero && cz <= limit; ++cz)
            p[cz][0][0] = 1;
        for (int co = 1; co <= one && co <= limit; ++co)
            p[0][co][1] = 1;
        for (int i = 0; i <= zero; i++) {
            for (int j = 0; j <= one; j++) {
                for (int last = 1; last <= limit; last++) {//全为tail的后缀数组的长度为last
                    if (i - last >= 0)
                        p[i][j][0] = (p[i][j][0] + p[i - last][j][1]) % mod;
                    if (j - last >= 0)
                        p[i][j][1] = (p[i][j][1] + p[i][j - last][0]) % mod;
                }
            }
        }
        return ((p[zero][one][0] + p[zero][one][1]) % mod + mod) % mod;
    }
};

D 找出所有稳定的二进制数组 II

动态规划:设 p [ i ] [ j ] [ t a i l ] p[i][j][tail] p[i][j][tail] 为含有 i i i 个 0 0 0 和 j j j 个 1 1 1 且以 t a i l tail tail 为结尾的稳定二进制数组的个数,枚举其全为 t a i l tail tail 的后缀数组的可能长度来进行状态转移,可以通过维护两个前缀和来优化状态转移的时间复杂度

cpp 复制代码
class Solution {
  public:
    using ll = long long;
    ll mod = 1e9 + 7;
    int numberOfStableArrays(int zero, int one, int limit) {
        ll p[zero + 1][one + 1][2]; 
        ll ps0[zero + 1][one + 1];
        ll ps1[zero + 1][one + 1];
        memset(p, 0, sizeof(p));
        memset(ps0, 0, sizeof(ps0));
        memset(ps1, 0, sizeof(ps1));
        for (int i = 1; i <= zero && i <= limit; ++i) {
            p[i][0][0] = 1;
            ps0[i][0] = 1;
        }
        for (int j = 1; j <= one && j <= limit; ++j) {
            p[0][j][1] = 1;
            ps1[0][j] = 1;
        }
        for (int i = 0; i <= zero; i++) {
            for (int j = 0; j <= one; j++) {
                // [max(0,i-limit),i-1]
                if (int l = max(0, i - limit), r = i - 1; l <= r)
                    p[i][j][0] += l != 0 ? (ps1[r][j] - ps1[l - 1][j]) % mod : ps1[r][j];
                if (int l = max(0, j - limit), r = j - 1; l <= r)
                    p[i][j][1] += l != 0 ? (ps0[i][r] - ps0[i][l - 1]) % mod : ps0[i][r];
                if (j)
                    ps0[i][j] = (ps0[i][j - 1] + p[i][j][0]) % mod;
                if (i)
                    ps1[i][j] = (ps1[i - 1][j] + p[i][j][1]) % mod;
            }
        }
        return ((p[zero][one][0] + p[zero][one][1]) % mod + mod) % mod;
    }
};
相关推荐
三维重建-光栅投影1 小时前
VS中将cuda项目编译为DLL并调用
算法
课堂剪切板3 小时前
ch03 部分题目思路
算法
山登绝顶我为峰 3(^v^)34 小时前
如何录制带备注的演示文稿(LaTex Beamer + Pympress)
c++·线性代数·算法·计算机·密码学·音视频·latex
Two_brushes.5 小时前
【算法】宽度优先遍历BFS
算法·leetcode·哈希算法·宽度优先
森焱森7 小时前
水下航行器外形分类详解
c语言·单片机·算法·架构·无人机
QuantumStack9 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法
写个博客10 小时前
暑假算法日记第一天
算法
绿皮的猪猪侠10 小时前
算法笔记上机训练实战指南刷题
笔记·算法·pta·上机·浙大
hie9889411 小时前
MATLAB锂离子电池伪二维(P2D)模型实现
人工智能·算法·matlab
杰克尼11 小时前
BM5 合并k个已排序的链表
数据结构·算法·链表