HDU 4107 Gangster Segment Tree線段樹

這道題也有點新意，就是需要記錄最小值段和最大值段，而後成段更新這個段，而不用沒點去更新，達到提快速度的目的。

本題過的人很是少，因爲大部分都超時了，我嚴格依照線段樹的方法去寫。一開始居然也超時。

而後修補了兩個地方就過了，詳細改動的地方請參看程序。

知道最大值段和最小值段，而後修補一下就能過了。不是特別難的題目。

#include <stdio.h>
#include <string>
#include <algorithm>
using namespace std;

const int SIZE = 200002;
const int TREESIZE = SIZE + (SIZE<<1);
int N, M, P;
int segTree[TREESIZE];
int smallest[TREESIZE];
int largest[TREESIZE];

inline int lChild(int rt) { return rt<<1; }
inline int rChild(int rt) { return rt<<1|1; }

void build(int l, int r, int rt)
{
	segTree[rt] = 0;
	smallest[rt] = 0;
	largest[rt] = 0;
	if (l == r) return ;

	int m = l + ((r-l)>>1);
	build(l, m, lChild(rt));
	build(m+1, r, rChild(rt));
}

inline void pushUp(int rt)
{
	smallest[rt] = min(smallest[lChild(rt)], smallest[rChild(rt)]);
	largest[rt] = max(largest[lChild(rt)], largest[rChild(rt)]);
}

inline void pushDown(int rt)
{
	if(segTree[rt])
	{
		int l = lChild(rt), r = rChild(rt);
		largest[l] += segTree[rt];
		largest[r] += segTree[rt];
		smallest[l] += segTree[rt];
		smallest[r] += segTree[rt];
		segTree[l] += segTree[rt];
		segTree[r] += segTree[rt];
		segTree[rt] = 0;
	}
}

void update(int L, int R, int val, int l, int r, int rt)
{
	if (L <= l && r <= R)
	{
		if(largest[rt] < P)
		{
			smallest[rt] += val;
			largest[rt] += val;
			segTree[rt] += val;
			return ;
		}
		if(smallest[rt] >= P)
		{
			smallest[rt] += val<<1;
			largest[rt] += val<<1;
			segTree[rt] += val<<1;
			return;
		}
	}

	//if (r < L || R < l) return;//在這裏推斷也會超時
	pushDown(rt);

	int m = l + ((r-l)>>1);
	if (L <= m) update(L, R, val, l, m, lChild(rt));
	if (m < R) update(L, R, val, m+1, r, rChild(rt));

	pushUp(rt);
}

inline void pushDown_2(int rt)
{
	if (segTree[rt])
	{
		segTree[lChild(rt)] += segTree[rt];
		segTree[rChild(rt)] += segTree[rt];
	}
}

void printTree(int l, int r, int rt)
{
	if (l == r)
	{
		if (l != N) printf("%d ", segTree[rt]);
		else printf("%d\n", segTree[rt]);
		return ;		//記得返回。
	}
	pushDown_2(rt);

	int m = l + ((r-l)>>1);
	printTree(l, m, lChild(rt));
	printTree(m+1, r, rChild(rt));
}

int main()
{
	int a, b, c;
	while (scanf("%d %d %d", &N, &M, &P) != EOF)
	{
		build(1, N, 1);
		//memset(largest, 0, sizeof(largest)); 這樣寫反而超時
		//memset(smallest, 0, sizeof(smallest));
		//memset(segTree, 0, sizeof(segTree));
		while (M--)
		{
			scanf("%d %d %d", &a, &b, &c);
			update(a, b, c, 1, N, 1);
		}
		printTree(1, N, 1);
	}
	return 0;
}