LeetCode第 34 场双周赛

5491. 矩阵对角线元素的和

给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。

示例 1：

输入：mat = [[1,2,3],
[4,5,6],
[7,8,9]]
输出：25
解释：对角线的和为：1 + 5 + 9 + 3 + 7 = 25
请注意，元素 mat[1][1] = 5 只会被计算一次。
示例 2：

输入：mat = [[1,1,1,1],
[1,1,1,1],
[1,1,1,1],
[1,1,1,1]]
输出：8
示例 3：

输入：mat = [[5]]
输出：5

提示：

n == mat.length == mat[i].length
1 <= n <= 100
1 <= mat[i][j] <= 100

代码

class Solution {
   
    public int diagonalSum(int[][] mat) {
   
        int n = mat.length, ret = 0;
        for (int i=0; i<n; ++i) {
   
            int j = n - 1 - i;
            ret += mat[i][i];
            if (i != j) {
   
                ret += mat[i][j];
            }
        }
        return ret;
    }
}

5492. 分割字符串的方案数

给你一个二进制串 s （一个只包含 0 和 1 的字符串），我们可以将 s 分割成 3 个 非空 字符串 s1, s2, s3 （s1 + s2 + s3 = s）。

请你返回分割 s 的方案数，满足 s1，s2 和 s3 中字符 ‘1’ 的数目相同。

由于答案可能很大，请将它对 10^9 + 7 取余后返回。

示例 1：

输入：s = “10101”
输出：4
解释：总共有 4 种方法将 s 分割成含有 ‘1’ 数目相同的三个子字符串。
“1|010|1”
“1|01|01”
“10|10|1”
“10|1|01”
示例 2：

输入：s = “1001”
输出：0
示例 3：

输入：s = “0000”
输出：3
解释：总共有 3 种分割 s 的方法。
“0|0|00”
“0|00|0”
“00|0|0”
示例 4：

输入：s = “100100010100110”
输出：12

提示：

s[i] == ‘0’ 或者 s[i] == ‘1’
3 <= s.length <= 10^5

思路

统计两个分割之间的0的个数。注意所有字符均为’0’的特殊情况。

代码

class Solution {
   
    private static final long mod = 1000000007;
    
    public int numWays(String s) {
   
        int cnt = 0;
        for (char ch: s.toCharArray()) {
   
            if (ch == '1') {
   
                ++cnt;
            }