leetcode剑指offer
552. 学生出勤记录 II
可以用字符串表示一个学生的出勤记录,其中的每个字符用来标记当天的出勤情况(缺勤、迟到、到场)。记录中只含下面三种字符:
‘A’:Absent,缺勤
‘L’:Late,迟到
‘P’:Present,到场
如果学生能够 同时 满足下面两个条件,则可以获得出勤奖励:
按 总出勤 计,学生缺勤(‘A’)严格 少于两天。
学生 不会 存在 连续 3 天或 连续 3 天以上的迟到(‘L’)记录。
给你一个整数 n ,表示出勤记录的长度(次数)。请你返回记录长度为 n 时,可能获得出勤奖励的记录情况 数量 。答案可能很大,所以返回对 109 + 7 取余 的结果。
示例 1:
输入:n = 2
输出:8
解释:
有 8 种长度为 2 的记录将被视为可奖励:
“PP” , “AP”, “PA”, “LP”, “PL”, “AL”, “LA”, “LL”
只有"AA"不会被视为可奖励,因为缺勤次数为 2 次(需要少于 2 次)。
示例 2:
输入:n = 1
输出:3
示例 3:
输入:n = 10101
输出:183236316
class Solution {
public:
int mod=1e9+7;
int checkRecord(int n) {
vector<vector<vector<int>>> dp(n+1,vector< vector<int> >(2,vector<int>(3,0)));
//dp[n][0|1][0]+=dp[n-1][0|1][0-2] //P
//dp[n][1][0]+=dp[n-1][0][0-2] //A
//dp[n][0|1][x+1]+=dp[n-1][0|1][x] //L x>=0&&x<=1
dp[1][0][0]=1;
dp[1][1][0]=1;
dp[1][0][1]=1;
for(int i=2;i<=n;i++){
for(int j=0;j<=1;j++){
for(int k=0;k<=2;k++){
dp[i][j][0]=(dp[i][j][0]+dp[i-1][j][k])%mod;
}
for(int k=0;k<=1;k++){
dp[i][j][k+1]=(dp[i][j][k+1]+dp[i-1][j][k])%mod;
}
}
for(int k=0;k<=2;k++){
dp[i][1][0]=(dp[i][1][0]+dp[i-1][0][k])%mod;
}
}
int ans=0;
for(int j=0;j<=1;j++){
for(int k=0;k<=2;k++){
ans=(ans+dp[n][j][k])%mod;
}
}
return ans;
}
};
第一感觉就是动归,因为需要枚举每一种情况,写这个状态转移的时候回忆起了写状态机的时候。
第二是用dfs写,我觉得dfs和动归两者核心思想是一样的。
第三是可以用矩阵快速幂优化。
不过我看了下,我和别人写的差不多,为什么我的代码跑的这么慢。