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
∗
1
0
5
1<=n<=2*10^5
1<=n<=2∗105
1
<
=
q
<
=
2
∗
1
0
5
1<=q<=2*10^5
1<=q<=2∗105
接下来q行输入操作
分别是
0
0
0
l
l
l
r
r
r
w
w
w
表示1到r区间内的人本月涨了w元工资
1
1
1
l
l
l
r
r
r
询问1,r内的人迄今为止发了多少钱
数据保证
1
<
=
l
<
=
r
<
=
n
,
1
<
=
w
<
=
1
0
9
1<=l<=r<=n,1<=w<=10^9
1<=l<=r<=n,1<=w<=109
输出描述:
每次询问发钱过后
输出一个整数,表示区间发过多少工资,并且取模998244353。
示例1
输入
3 2
0 1 3 1000000000
1 1 3
输出
10533882
考虑一个人的工资发放情况,横轴为月份,纵轴为工资,
蓝色区域内的边界随着时间的增加而增加,因此不好计算它的贡献。而整个矩形区域的面积是好计算的,红色区域的边界固定,因此也容易计算。
所以每次修改时就区间减去红色区域的值即可。
红色区域的值 = ( t - 1) * 涨的工资数
最后答案就是整个矩形的面积减去红色区域的值。利用数据结构维护这两个值即可。时间复杂度为: O ( n l o g n ) (nlogn) (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();
}
}