1. 程式人生 > >nyoj 116 士兵殺敵(二)

nyoj 116 士兵殺敵(二)

士兵殺敵(二)

時間限制:1000 ms  |  記憶體限制:65535 KB 難度:5
描述

南將軍手下有N個士兵,分別編號1到N,這些士兵的殺敵數都是已知的。

小工是南將軍手下的軍師,南將軍經常想知道第m號到第n號士兵的總殺敵數,請你幫助小工來回答南將軍吧。

南將軍的某次詢問之後士兵i可能又殺敵q人,之後南將軍再詢問的時候,需要考慮到新增的殺敵數。

輸入
只有一組測試資料
第一行是兩個整數N,M,其中N表示士兵的個數(1<N<1000000),M表示指令的條數。(1<M<100000)
隨後的一行是N個整數,ai表示第i號士兵殺敵數目。(0<=ai<=100)
隨後的M行每行是一條指令,這條指令包含了一個字串和兩個整數,首先是一個字串,如果是字串QUERY則表示南將軍進行了查詢操作,後面的兩個整數m,n,表示查詢的起始與終止士兵編號;如果是字串ADD則後面跟的兩個整數I,A(1<=I<=N,1<=A<=100),表示第I個士兵新增殺敵數為A.
輸出
對於每次查詢,輸出一個整數R表示第m號士兵到第n號士兵的總殺敵數,每組輸出佔一行
樣例輸入
5 6
1 2 3 4 5
QUERY 1 3
ADD 1 2
QUERY 1 3
ADD 2 3
QUERY 1 2
QUERY 1 5
樣例輸出
6
8
8
20
這道題用樹狀陣列來做,並且需要注意用scanf,printf.cin,cout是無法通過的.
#include <cstdio>
#include <cstring>
using namespace std;
#define MAX_N 1000100
int a[MAX_N];
int n;
void update(int i, int x)
{
	while (i <= n)
	{
		a[i] += x;
		i += i&-i;
	}
}
int sum(int i)
{
	int s = 0;
	while (i > 0)
	{
		s += a[i];
		i -= i&-i;
	}
	return s;
}
int main()
{
	int c, d;
	int i;
	char s[10];
	while (~scanf("%d%d",&n,&i))
	{
		for (c = 1; c <= n; c++)
		{
			scanf("%d",&d);
			update(c, d);
		}
		while (i--)
		{
			scanf("%s%d%d", s, &c, &d);
			if (s[0]=='Q')
			{
				printf("%d\n",sum(d)-sum(c-1));
			}
			else
			{
				update(c, d);
			}
		}
	}
}