leetcode 552. Student Attendance Record II

最新推荐文章于 2021-08-20 01:22:23 发布

Hackerzer

最新推荐文章于 2021-08-20 01:22:23 发布

阅读量447

点赞数

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.csdn.net/hackerzer/article/details/74906734

算法专栏收录该内容

157 篇文章

订阅专栏

本文介绍了一种使用动态规划解决 LeetCode 552 题目（学生考勤记录 II）的方法。该题目要求计算在不超过限制条件下可能的考勤记录数。通过三维数组 f 来记录不同状态的数量，并最终返回所有可能的状态数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

leetcode 552. Student Attendance Record II

又一题动态规划

public class Solution {
    public int checkRecord(int n) {
        final int MOD = 1000000007;
        int[][][] f = new int[n+1][2][3];
        // at most 1 A, at most 2 L
        f[0] = new int[][]{{1, 1, 1}, {1, 1, 1}};
        for(int i=1;i<=n;i++){
            for(int j=0;j<=1;j++){
                for(int k=0;k<=2;k++){
                    int val = f[i-1][j][2]; // ..p
                    if(j>0) val = (val + f[i-1][j-1][2])%MOD;  // ..A
                    if(k>0) val = (val + f[i-1][j][k-1])%MOD;  // ..L
                    f[i][j][k] = val;  //
                }
            }
        }
        return f[n][1][2];
    }
}