leetcode 记录

2022-11-10 21:26:24 浏览数 (1)

题目:(题号1622)

请你实现三个 API append,addAll 和 multAll 来实现奇妙序列。

请实现 Fancy 类 :

Fancy() 初始化一个空序列对象。 void append(val) 将整数 val 添加在序列末尾。 void addAll(inc) 将所有序列中的现有数值都增加 inc 。 void multAll(m) 将序列中的所有现有数值都乘以整数 m 。 int getIndex(idx) 得到下标为 idx 处的数值(下标从 0 开始),并将结果对 109 7 取余。如果下标大于等于序列的长度,请返回 -1 。

示例:

输入: [“Fancy”, “append”, “addAll”, “append”, “multAll”, “getIndex”, “addAll”, “append”, “multAll”, “getIndex”, “getIndex”, “getIndex”] [[], [2], [3], [7], [2], [0], [3], [10], [2], [0], [1], [2]] 输出:

[null, null, null, null, null, 10, null, null, null, 26, 34, 20]

解释: Fancy fancy = new Fancy(); fancy.append(2); // 奇妙序列:[2] fancy.addAll(3); // 奇妙序列:[2 3] -> [5] fancy.append(7); // 奇妙序列:[5, 7] fancy.multAll(2); // 奇妙序列:[52, 72] -> [10, 14] fancy.getIndex(0); // 返回 10 fancy.addAll(3); // 奇妙序列:[10 3, 14 3] -> [13, 17] fancy.append(10); // 奇妙序列:[13, 17, 10] fancy.multAll(2); // 奇妙序列:[132, 172, 10*2] -> [26, 34, 20] fancy.getIndex(0); // 返回 26 fancy.getIndex(1); // 返回 34 fancy.getIndex(2); // 返回 20

提示:

1 <= val, inc, m <= 100 0 <= idx <= 105 总共最多会有 105 次对 append,addAll,multAll 和 getIndex 的调用。

题解:

代码语言:javascript复制
class Fancy {

    private final long MOD = (long) 1e9   7;

    private long mul, add;

    private List nums;

    private List addList, mulList;

    public Fancy() {
        nums = new ArrayList<>();
        addList = new ArrayList<>();
        mulList = new ArrayList<>();
        mul = 1;
        add = 0;
    }

    public void append(int val) {
        nums.add(val);
        addList.add(add);
        mulList.add(mul);
    }

    public void addAll(int inc) {
        add = (add   inc) % MOD;
    }

    public void multAll(int m) {
        add = (add * m) % MOD;
        mul = (mul * m) % MOD;
    }

    public int getIndex(int idx) {
        if (idx >= nums.size()) return -1;
        long ans = nums.get(idx);
        long q = qpow(mulList.get(idx), MOD - 2L);
        ans = ans * mul % MOD;
        ans = ans * q % MOD;
        ans = (ans   add) % MOD;
        ans = (ans - addList.get(idx) * mul % MOD * q % MOD   MOD) % MOD;
        return (int) ans;
    }

    private long qpow(long a, long p) {
        long ans = 1;
        while (p > 0) {
            if ((p & 1) > 0) ans = ans * a % MOD;
            a = a * a % MOD;
            p >>= 1;
        }
        return ans;
    }
}

0 人点赞