3687. 图书馆逾期罚款计算器 🔒

English Version

题目描述

给定一个整数数组 daysLate，其中 daysLate[i] 表示第 i 书的归还日期晚了几天。

罚款计算如下：

• 如果 daysLate[i] == 1，罚款为 1。
• 如果 2 <= daysLate[i] <= 5，罚款为 2 * daysLate[i]。
• 如果 daysLate[i] > 5，罚款为 3 * daysLate[i]。

返回所有书的总罚款。

 

示例 1：

示例 2：

 

提示：

• 1 <= daysLate.length <= 100
• 1 <= daysLate[i] <= 100

解法

方法一：模拟

我们定义一个函数 $\text{f}(x)$ 来计算每本书的罚款：

$$\text{f}(x) = \begin{cases} 1 & x = 1 \\ 2x & 2 \leq x \leq 5 \\ 3x & x > 5 \end{cases}$$

然后我们对数组 $\textit{daysLate}$ 中的每个元素 $x$ 计算 $\text{f}(x)$ 并累加得到总罚款。

时间复杂度 $O(n)$，其中 $n$ 是数组 $\textit{daysLate}$ 的长度。空间复杂度 $O(1)$。

Python3

class Solution:
    def lateFee(self, daysLate: List[int]) -> int:
        def f(x: int) -> int:
            if x == 1:
                return 1
            if x > 5:
                return 3 * x
            return 2 * x

        return sum(f(x) for x in daysLate)

Java

class Solution {
    public int lateFee(int[] daysLate) {
        IntUnaryOperator f = x -> {
            if (x == 1) {
                return 1;
            } else if (x > 5) {
                return 3 * x;
            } else {
                return 2 * x;
            }
        };

        int ans = 0;
        for (int x : daysLate) {
            ans += f.applyAsInt(x);
        }
        return ans;
    }
}

C++

class Solution {
public:
    int lateFee(vector<int>& daysLate) {
        auto f = [](int x) {
            if (x == 1) {
                return 1;
            } else if (x > 5) {
                return 3 * x;
            } else {
                return 2 * x;
            }
        };

        int ans = 0;
        for (int x : daysLate) {
            ans += f(x);
        }
        return ans;
    }
};

Go

func lateFee(daysLate []int) (ans int) {
	f := func(x int) int {
		if x == 1 {
			return 1
		} else if x > 5 {
			return 3 * x
		}
		return 2 * x
	}
	for _, x := range daysLate {
		ans += f(x)
	}
	return
}

TypeScript

function lateFee(daysLate: number[]): number {
    const f = (x: number): number => {
        if (x === 1) {
            return 1;
        } else if (x > 5) {
            return 3 * x;
        }
        return 2 * x;
    };
    return daysLate.reduce((acc, days) => acc + f(days), 0);
}