题目描述
小新经常陪小白去公园玩,也就是所谓的遛狗啦…在小新家附近有一条“公园路”,路的一边从南到北依次排着n个公园,小白早就看花了眼,自己也不清楚该去哪些公园玩了。
一开始,小白就根据公园的风景给每个公园打了分-.-。小新为了省事,每次遛狗的时候都会事先规定一个范围,小白只可以选择第a个和第b个公园之间(包括a、b两个公园)选择连续的一些公园玩。小白当然希望选出的公园的分数总和尽量高咯。同时,由于一些公园的景观会有所改变,所以,小白的打分也可能会有一些变化。
那么,就请你来帮小白选择公园吧。
输入格式
第一行,两个整数N和M,分别表示表示公园的数量和操作(遛狗或者改变打分)总数。
接下来N行,每行一个整数,依次给出小白 开始时对公园的打分。
接下来M行,每行三个整数。第一个整数K,1或2。K=1表示,小新要带小白出去玩,接下来的两个整数a和b给出了选择公园的范围(1≤a,b≤N);K=2表示,小白改变了对某个公园的打分,接下来的两个整数p和s,表示小白对第p个公园的打分变成了s(1≤p≤N)。
其中,1≤N≤500 000,1≤M≤100 000,所有打分都是绝对值不超过1000的整数。
输出格式
小白每出去玩一次,都对应输出一行,只包含一个整数,表示小白可以选出的公园得分和的最大值。
样例输入
5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 2 3
样例输出
2
-1
一道線段樹的入門題。用四個標記維護每個節點的狀態:
tree[p].maxL表示該區間內包含最左端元素的最大和;
tree[p].maxR表示該區間內包含最右端元素的最大和;
tree[p].sum表示该区间的綜合;
tree[p].maxS表示該區間的最大子區間和。
那麼每個p節點的狀態與其左子樹lc和右子樹rc的關係為:
tree[p].maxL = max(lc.maxL, lc.sum + rc.maxL);
tree[p].maxR = max(rc.maxR, rc.sum + lc.maxR);
tree[p].sum = lc.sum + rc.sum;
tree[p].maxS = max(lc.maxS, rc.maxS,
lc.maxR + rc.maxL)。
具體細節建程序。
Accode:
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <bitset>
using std::max;
const char fi[] = "rqnoj626.in";
const char fo[] = "rqnoj626.out";
const int maxN = 500010;
const int maxM = 100010;
const int MAX = 0x3fffff00;
const int MIN = -MAX;
struct SegTree
{
int L, R, lc, rc;
int maxL, maxR, sum, maxS;
};
SegTree tree[maxN << 2];
int sc[maxN];
int n, m, tot;
void init_file()
{
freopen(fi, "r", stdin);
freopen(fo, "w", stdout);
}
void update(SegTree *ths, SegTree *lc, SegTree *rc)
{
ths -> maxL = max(lc -> maxL,
lc -> sum + rc -> maxL);
ths -> maxR = max(rc -> maxR,
rc -> sum + lc -> maxR);
ths -> sum = lc -> sum + rc -> sum;
ths -> maxS = max(lc -> maxR + rc -> maxL,
max(lc -> maxS, rc -> maxS));
} //根據左子樹lc和右子樹rc更新ths節點的狀態。
void Build(int L, int R)
{
int Now = ++tot;
tree[Now].L = L;
tree[Now].R = R;
if (L == R)
{
tree[Now].maxL = tree[Now].maxR
= tree[Now].sum = tree[Now].maxS
= sc[L];
return;
}
//若到葉節點則直接賦初始打分並返回。
int Mid = (L + R) >> 1;
if (L < R)
{
tree[Now].lc = tot + 1;
Build(L, Mid);
//先建左子樹。
tree[Now].rc = tot + 1;
Build(Mid + 1, R);
//後建右子樹。
}
update(tree + Now, tree + tree[Now].lc,
tree + tree[Now].rc);
//根據左右子樹的信息更新當前節點。
}
void Modify(int Now, int p)
{
if (tree[Now].L == tree[Now].R
&& tree[Now].L == p)
{
tree[Now].maxL = tree[Now].maxR
= tree[Now].sum = tree[Now].maxS
= sc[p];
return;
}
//邊界,若找到p則直接修改。
int Mid = (tree[Now].L + tree[Now].R) >> 1;
if (p <= Mid) Modify(tree[Now].lc, p);
//若p在左區間則修改左子樹。
if (p > Mid) Modify(tree[Now].rc, p);
//若p在右區間則修改右子樹。
update(tree + Now, tree + tree[Now].lc,
tree + tree[Now].rc);
//更新當前節點。
}
SegTree Find(int Now, int L, int R)
{
if (L <= tree[Now].L && R >= tree[Now].R)
return tree[Now];
//若被完全覆蓋則直接返回樹上當前節點的狀態。
int Mid = (tree[Now].L + tree[Now].R) >> 1;
if (R <= Mid)
return Find(tree[Now].lc, L, R);
//若完全被左子樹所在區間覆蓋,
//則直接返回左子樹的狀態。
if (L > Mid)
return Find(tree[Now].rc, L, R);
//同理單獨處理右子樹。
SegTree ans;
SegTree lc = Find(tree[Now].lc, L, R);
//找到左子樹的狀態。
SegTree rc = Find(tree[Now].rc, L, R);
//找到右子樹的狀態。
update(&ans, &lc, &rc);
//更新當前節點的狀態。
return ans;
} //返回在Now節點,區間[L, R]中找到最優狀態。
void work()
{
scanf("%d%d", &n, &m);
for (int i = 1; i < n + 1; ++i)
scanf("%d", sc + i);
tot = 0;
Build(1, n);
//建樹並初始化各節點的狀態(根據最初打分)。
for (; m; --m)
{
int O, x, y;
scanf("%d%d%d", &O, &x, &y);
switch (O)
{
case 1 :
if (x > y) std::swap(x, y);
//注意有可能x < y。
printf("%d\n", Find(1, x, y).maxS);
break;
case 2 :
if (sc[x] == y) break;
sc[x] = y;
Modify(1, x);
break;
}
}
}
int main()
{
init_file();
work();
exit(0);
}