2023河南萌新联赛第（三）场：郑州大学 A - 发工资咯

2023河南萌新联赛第（三）场：郑州大学 A - 发工资咯

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

一个公司有n个人，每个月都要给这n个人发工资，刚开始每个人的工资都为0，每月底公司会进行两种操作。

1.挑一段连续的区间给区间内的人涨工资

2.询问某个区间内的人迄今为止已经发了多少工资。

请你回答每个操作2

输入描述:

第一行输入n，q
$1<=n<=2\ast 10^5$ $1<=q<=2\ast 10^5$
接下来q行输入操作
分别是
$0$ $l$ $r$ $w$
表示1到r区间内的人本月涨了w元工资
$1$ $l$ $r$
询问1，r内的人迄今为止发了多少钱
数据保证$1<=l<=r<=n，1<=w<=10^9$

输出描述:

每次询问发钱过后

输出一个整数，表示区间发过多少工资，并且取模998244353。

示例1

输入

3 2
0 1 3 1000000000
1 1 3

输出

10533882

考虑一个人的工资发放情况，横轴为月份，纵轴为工资，

蓝色区域内的边界随着时间的增加而增加，因此不好计算它的贡献。而整个矩形区域的面积是好计算的，红色区域的边界固定，因此也容易计算。

所以每次修改时就区间减去红色区域的值即可。

红色区域的值 = ( t - 1) * 涨的工资数

最后答案就是整个矩形的面积减去红色区域的值。利用数据结构维护这两个值即可。时间复杂度为： O$(nlogn)$

import java.io.*;

public class Main {
    static int MOD = 998244353;
    static int N = 200010;
    static int[] c1 = new int[N];
    static int[] c2 = new int[N];
    static int[] d1 = new int[N];
    static int[] d2 = new int[N];
    static int n, q;

    public static void add(int x, int v, int[] tr1, int[] tr2) {
        for (int i = x; i <= n; i += (i & -i)) {
            tr1[i] = (tr1[i] + v) % MOD;
            tr2[i] = (int) (tr2[i] + ((long) (x - 1) * v) % MOD) % MOD;
        }
    }

    public static long query(int x, int[] tr1, int[] tr2) {
        long res = 0;
        for (int i = x; i > 0; i -= (i & -i)) {
            res = (res + ((long) x * tr1[i] % MOD - tr2[i] + MOD)) % MOD;
        }
        return res;
    }

    public static void main(String[] args) throws IOException {
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        String[] str = bf.readLine().split(" ");
        n = Integer.parseInt(str[0]);
        q = Integer.parseInt(str[1]);
        for (int i = 1; i <= q; i++) {
            str = bf.readLine().split(" ");
            int op = Integer.parseInt(str[0]);
            int l = Integer.parseInt(str[1]);
            int r = Integer.parseInt(str[2]);
            if (op == 0) {
                int w = Integer.parseInt(str[3]);
                add(l, w % MOD, c1, c2);
                add(r + 1, (MOD - w + MOD) % MOD, c1, c2);
                long t = q - i + 1;
                long dd1 = t * w % MOD;
                add(l, (int) dd1, d1, d2);
                long dd2 = t * (MOD - w + MOD) % MOD;
                add(r + 1, (int) dd2, d1, d2);
            } else {
                long ans = query(r, c1, c2) - query(l - 1, c1, c2);
                ans = (ans + MOD) % MOD;
                ans = ans * (q - i) % MOD;
                long res = query(r, d1, d2) - query(l - 1, d1, d2);
                res = (res + MOD) % MOD;
                res = (res - ans + MOD) % MOD;
                bw.write(res % MOD + "\n");
            }
        }
        bw.close();
    }
}