站外题求助
  • 板块学术版
  • 楼主ZettaByte
  • 当前回复4
  • 已保存回复4
  • 发布时间2024/2/29 22:27
  • 上次更新2024/2/29 23:34:54
查看原帖
站外题求助
685260
ZettaByte楼主2024/2/29 22:27

题面:

线段树,不知道为什么WA了四个点,希望有大佬来帮忙调一下。

代码:

#include <bits/stdc++.h>
using namespace std;
const int N = 200010;
int a[N], tr[N<<2], tg[N<<2];
int ls(int p) { return p << 1; }
int rs(int p) { return p << 1 | 1; }
void up(int p) { tr[p] = max(tr[ls(p)], tr[rs(p)]); }
void build(int p, int pl, int pr)
{
	tg[p] = 0;
	if (pl == pr) { tr[p] = a[pl]; return ; }
	int mid = (pl + pr) >> 1; build(ls(p), pl, mid), build(rs(p), mid + 1, pr), up(p);
}
void addtg(int p, int pl, int pr, int d) { tg[p] += d, tr[p] += d; }
void down(int p, int pl, int pr)
{
	if (tg[p])
	{
		int mid = (pl + pr) >> 1;
		addtg(ls(p), pl, mid, tg[p]), addtg(rs(p), mid + 1, pr, tg[p]), tg[p] = 0;
	}
}
void add(int l, int r, int p, int pl, int pr, int d)
{
	if (l <= pl && pr <= r) { addtg(p, pl, pr, d); return ; }
	down(p, pl, pr); int mid = (pl + pr) >> 1;
	if (l <= mid) add(l, r, ls(p), pl, mid, d);
	if (r > mid) add(l, r, rs(p), mid + 1, pr, d);
	up(p);
}
int ask(int l, int r, int p, int pl, int pr)
{
	if (l <= pl && pr <= r) return tr[p];
	down(p, pl, pr); int ret = 0, mid = (pl + pr) >> 1;
	if (l <= mid) ret = max(ret, ask(l, r, ls(p), pl, mid));
	if (r > mid) ret = max(ret, ask(l, r, rs(p), mid + 1, pr));
	return ret;
}
void upd_1(int rt, int l, int r, int idx, int add)
{
	if (l == r) { tr[rt] += add; return ; }
	int mid = (l + r) / 2;
	if (idx <= mid) upd_1(rt * 2, l, mid, idx, add);
	else upd_1(rt * 2 + 1, mid + 1, r, idx, add);
	up(rt);
}

int main()
{
//	freopen("score.in", "r", stdin);
//	freopen("score.out", "w", stdout);
	int T; scanf("%d", &T);
	while (T--)
	{
		memset(a, 0, sizeof(a)), memset(tr, 0, sizeof(tr)), memset(tg, 0, sizeof(tg));
		int n, m; scanf("%d%d", &n, &m);
		for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
		build(1, 1, n);
		while (m--)
		{
			int l, r; char op; scanf("\n%c %d %d", &op, &l, &r);
			if (op == 'U') upd_1(1, 1, n, l, r - l);
			else printf("%d\n", ask(l, r, 1, 1, n));
		}
	}
	return 0;
}
2024/2/29 22:27
加载中...