6250. 商店的最少代价

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第6天，点击查看活动详情

题目描述

给你一个顾客访问商店的日志，用一个下标从 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 。

解题思路

如果当前关门就需要考虑之前的N后之后的Y，将两部分加起来就是总代价，再去从下标0开始到n结束比较出最小代价。先计算出总的y数量，循环的时候再记录已经出现的n的数量，这样每一步的代价=之前n的数量与之后y的数量相加。注意最后要再判断如果在最后再关门的情况

代码

class Solution {
    public int bestClosingTime(String customers) {
        int n = customers.length();
        int yNum = 0;
        // 找到Y的总数
        for (int i = 0; i < n; i++) {
            if (customers.charAt(i) == 'Y') {
                yNum ++;
            }
        }
        int beforeNNum = 0;
        int ans = n;
        int ansNum = n;
        for (int i = 0; i < n; i++) {
            // 当前关门代价等于 之后y的数量+之前n的数量
            int num = yNum + beforeNNum;
            if (num < ansNum) {
                ansNum = num;
                ans = i;
            }
            if (customers.charAt(i) == 'Y') {
                yNum --;
            } else {
                beforeNNum ++;
            }
        }
        if (beforeNNum < ansNum) {
            return n;
        }
        return ans;
    }
}

总结

本题主要就是注意通过记录前面出现n的次数来简便计算。