题目链接
题目描述
可以用字符串表示一个学生的出勤记录,其中的每个字符用来标记当天的出勤情况(缺勤、迟到、到场)。记录中只含下面三种字符:
‘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
提示:
1 <= n <= 105
求解思路
- 深度优先搜索:
cache
数组用于缓存每次深搜得到的结果,cache[i][j[k]
表示剩余i
个数字需要决策时,A出现j
次,L出现k
次的情况次数。 public int dfs(int t, int Acnt, int Lcnt)
方法中,t
表示剩余需要决策的次数,Acnt
和Lcnt
分别表示缺勤次数和结尾出迟到的连续次数。
实现代码
class Solution {
final int mod = (int)1e9 + 7;
int[][][] cache;
public int checkRecord(int n) {
cache = new int[n + 1][2][3];
for (int i = 0; i <= n; i ++) {
for (int j = 0; j < 2; j ++) {
for (int k = 0; k < 3; k ++) {
cache[i][j][k] = -1;
}
}
}
return dfs(n, 0, 0);
}
public int dfs(int t, int Acnt, int Lcnt) {
if (Acnt >= 2 || Lcnt >= 3) {
return 0;
} else if (t == 0) {
return 1;
} else if (cache[t][Acnt][Lcnt] != -1) {
return cache[t][Acnt][Lcnt];
}
int ans = 0;
ans = dfs(t - 1, Acnt + 1, 0) % mod; //A
ans = (ans + dfs(t - 1, Acnt, Lcnt + 1)) % mod; //L
ans = (ans + dfs(t - 1, Acnt, 0)) % mod; //P
cache[t][Acnt][Lcnt] = ans;
return ans;
}
}