NYOJ-士兵杀敌（五） 线段树区间更新、线段求和

士兵杀敌（五）

时间限制： 2000 ms  |  内存限制： 65535 KB

难度： 5

描述

南将军麾下有百万精兵，现已知共有M个士兵，编号为0~M，每次有任务的时候，总会有一批编号连在一起人请战（编号相近的人经常在一块，相互之间比较熟悉），最终他们获得的军功，也将会平分到每个人身上，这样，有时候，计算他们中的哪一个人到底有多少军功就是一个比较困难的事情。

在这样的情况下，南将军却经常会在许多次战役之后询问军师小工第i号士兵到第j号士兵所有人的总军功数。

请你帮助军师小工回答南将军的提问。

输入

只有一组测试数据
第一行是三个整数N,C,Q(1<=N,C,Q<=1000000)，其中N表示士兵的总数。
随后的C行，每行有三个整数Mi,Ni,Ai(0<=Mi<=Ni<=N,0<=Ai<=100)，表示从第Mi号到第Ni号士兵所有人平均增加了Ai的军功。
再之后的Q行，每行有两个正正数m,n，表示南将军询问的是第m号士兵到第n号士兵。

输出

请对每次询问输出m号士兵到第n号士兵的总军功数，由于该数值可能太大，请把结果对10003取余后输出

样例输入

5 3 2
1 3 2
2 4 1
5 5 10
1 5
2 3

样例输出

19
6

解题思路：

这题使用线段树来写，妥妥的超时了，然后使用了前缀和来写，过了。直接上代码

线段树代码（超时）

#include <cstdio>
#include <iostream>
using namespace std;
#define Max 1000005
#define ll long long
ll sum[Max<<2]; //记录每个节点的sum值 
ll co[Max<<2];//延迟标记数组 
void Build(int l, int r, int rt)
{
	co[rt] = 0;//标记初始全部置0 
	sum[rt] = 0;
	if(l == r) {
	//	scanf("%lld", &sum[rt]);
		return ;
	}
	int m = (l + r) >>1;
	Build(l, m, rt<<1);
	Build(m+1, r, rt<<1|1);
	sum[rt] = sum[rt<<1] + sum[rt<<1|1];//递归得到非叶子节点的sum值 
}
void Updown(int rt, int key)//向下标记函数 
{
	if(co[rt])//如果有标记，即下面的区间需要处理 
	{
		co[rt<<1] += co[rt];  //注意这里是相加，不是赋值 
		co[rt<<1|1] += co[rt];  //hdu这里是赋值，因为那是区间替换 
		sum[rt<<1] += (key - (key >> 1)) * co[rt];//左子树sum值相加更新 
		sum[rt<<1|1] += (key >> 1) * co[rt];//右子树sum值相加更新
		co[rt]=0;//标记处理完后要清为0，因为更新已经结束，任务结束 
	}
}
void Updata(int L, int R, int d, int l, int r, int rt)//区间更新函数 
{
	if(l >= L && r <= R)
	{
		co[rt] += d; //标记，为后面查询做准备 
		sum[rt] += (ll)d * (r - l +1);//该节点需要在这里处理，因为已经是最下面的了，不会再下去了 
		return ;
	}
	Updown(rt, r - l + 1);//向下延迟标记 
	int m = (l + r) >> 1;
	if(L <= m)  Updata(L, R, d, l, m, rt<<1);  //这两行能节省大量时间 
	if(m < R)  Updata(L, R, d, m+1, r, rt<<1|1);// 标准写法，代码简介 
	sum[rt] = sum[rt<<1] + sum[rt<<1|1];//回溯更新当前sum值 
}
ll Quest(int L, int R, int l, int r, int rt)//查询函数 
{
   if(l >= L && r <= R)
   {
   	return sum[rt];
   }
   Updown(rt, r - l + 1);//必须要这一行，向下更新，很重要 
   int m = (l + r) >> 1;
   ll ret = 0;
	if(L <= m)  ret += Quest(L, R, l, m, rt<<1) % 10003;//这两行递归求解和 
	if(m < R)  ret += Quest(L, R, m+1, r, rt<<1|1) % 10003;
	return ret % 10003;
}
int main()
{
	int N, Q ,T;
	scanf("%d%d%d", &N, &Q, &T);
	Build(1, N, 1);
	int a,b,d;
	while(Q--)
	{
	   scanf("%d%d%d", &a, &b, &d);
		Updata(a, b, d, 1, N, 1);
	}
	while(T--)		  
	{
		scanf("%d%d", &a, &b);
		 printf("%lld\n", Quest(a, b, 1, N, 1) % 10003);
	}
		return 0;
}

前缀和代码（ac）

#include <cstdio>
#include <cstring>
int f[1000005];
int main()
{
	int n, c, q, i;
	int a,b,cc;
	scanf("%d%d%d", &n, &c, &q);
	memset(f,0,sizeof(f));
	for(i = 0; i < c; i++)
	{
		scanf("%d%d%d", &a, &b, &cc);
		f[a] += cc;
		f[b+1] -= cc;
	}
	for(i = 1; i <= n; i++)
	  f[i] += f[i-1];
	for(i = 1; i <= n; i++)
	  f[i] = (f[i] + f[i-1]) % 10003;
	while(q--)
	{
		scanf("%d%d", &a, &b);
		printf("%d\n", (f[b] - f[a-1] + 10003) % 10003);
	}
	return 0;
}

题目链接：点击打开链接http://acm.nyist.net/JudgeOnline/problem.php?pid=228