1. 程式人生 > >1698 Just a Hook 線段樹區間更新

1698 Just a Hook 線段樹區間更新

    線段樹區間更新,有個懶惰標記,每次更新不進行到底,,用懶惰標記使得更新延遲到下次需要更新或者詢問到的時候。線段樹的精髓在於能不向下更新,儘可能利用子樹的根的資訊去獲取整棵子樹的資訊。

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>

#define lson l, m, rt << 1
#define rson m+1, r, rt << 1 | 1

using namespace std;
const int MAXN = 200020;

int col[MAXN<<2];//懶惰標記,每次更新到下一次需要更新的或者需要詢問的節點
int sum[MAXN<<2];

void PushUp(int rt)
{
	sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}

void PushDown(int rt, int m)//鄉下更新一次,與懶惰標記一起理解
{
	if(col[rt])//col[rt]表明上一次已經更新到rt節點,現在需要向其子節點更新一次
	{
		col[rt<<1] = col[rt<<1|1] = col[rt];
		sum[rt<<1] = col[rt] * (m - (m>>1));
		sum[rt<<1|1] = col[rt] * (m>>1);
		col[rt] = 0;
	}
}

void Build(int l, int r, int rt)
{
	col[rt] = 0;
	sum[rt] = 1;
	if(l == r)
		return;
	int m = (l+r)>>1;
	Build(lson);
	Build(rson);
	PushUp(rt);
}

void UpData(int L, int R, int c, int l, int r, int rt)
{
	if(L <= l && R >= r)
	{
		col[rt] = c;
		sum[rt] = c * (r-l+1);
		return;
	}
	PushDown(rt, r-l+1);
	int m = (l+r)>>1;
	if(L <= m)
		UpData(L, R, c, lson);
	if(R > m)
		UpData(L, R, c, rson);
	PushUp(rt);
}

int main()
{
	int T, n, Q;
	int kcase = 1;
	scanf("%d", &T);
	while(T--)
	{
		scanf("%d%d", &n, &Q);
		Build(1, n, 1);
		while(Q--)
		{
			int a, b, c;
			scanf("%d%d%d",&a, &b, &c);
			UpData(a, b, c, 1, n, 1);
		}
		printf("Case %d: The total value of the hook is %d.\n",kcase++, sum[1]);
	}
	system("pause");
	return 0;
}