目标
给你一个顾客访问商店的日志,用一个下标从 0 开始且只包含字符 'N' 和 'Y' 的字符串 customers 表示:
- 如果第 i 个字符是 'Y' ,它表示第 i 小时有顾客到达。
- 如果第 i 个字符是 'N' ,它表示第 i 小时没有顾客到达。
如果商店在第 j 小时关门(0 <= j <= n),代价按如下方式计算:
- 在开门期间,如果某一个小时没有顾客到达,代价增加 1 。
- 在关门期间,如果某一个小时有顾客到达,代价增加 1 。
请你返回在确保代价 最小 的前提下,商店的 最早 关门时间。
注意,商店在第 j 小时关门表示在第 j 小时以及之后商店处于关门状态。
示例 1:
输入:customers = "YYNY"
输出:2
解释:
- 第 0 小时关门,总共 1+1+0+1 = 3 代价。
- 第 1 小时关门,总共 0+1+0+1 = 2 代价。
- 第 2 小时关门,总共 0+0+0+1 = 1 代价。
- 第 3 小时关门,总共 0+0+1+1 = 2 代价。
- 第 4 小时关门,总共 0+0+1+0 = 1 代价。
在第 2 或第 4 小时关门代价都最小。由于第 2 小时更早,所以最优关门时间是 2 。
示例 2:
输入:customers = "NNNNN"
输出:0
解释:最优关门时间是 0 ,因为自始至终没有顾客到达。
示例 3:
输入:customers = "YYYY"
输出:4
解释:最优关门时间是 4 ,因为每一小时均有顾客到达。
说明:
- 1 <= customers.length <= 10^5
- customers 只包含字符 'Y' 和 'N' 。
思路
有一个顾客访问日志 customers,customers[i] 表示在第 i 小时是否有顾客到达,关店代价的计算方式为,营业期间没有顾客到达或者关店后(包括关店的时刻)有顾客到达代价 +1。返回代价最小的最早关门时间。
前后缀分解。
代码
/**
* @date 2025-12-26 8:57
*/
public class BestClosingTime2483 {
public int bestClosingTime(String customers) {
int n = customers.length();
int[] prefix = new int[n + 1];
int[] suffix = new int[n + 1];
for (int i = 0; i < n; i++) {
prefix[i + 1] = prefix[i] + (customers.charAt(i) == 'N' ? 1 : 0);
suffix[n - i - 1] = suffix[n - i] + (customers.charAt(n - i - 1) == 'Y' ? 1 : 0);
}
int min = Integer.MAX_VALUE;
int res = 0;
for (int i = 0; i <= n; i++) {
int cost = prefix[i] + suffix[i];
if (min > cost) {
res = i;
min = cost;
}
}
return res;
}
}
性能
