hihocoder #1077 RMQ問題再臨-線段樹

#1077 : RMQ問題再臨-線段樹

時間限制:10000ms

單點時限:1000ms

內存限制:256MB

輸入

每個測試點（輸入文件）有且僅有一組測試數據。

每組測試數據的第1行爲一個整數N，意義如前文所述。

每組測試數據的第2行爲N個整數，分別描述每種商品的重量，其中第i個整數表示標號爲i的商品的重量weight_i。

每組測試數據的第3行爲一個整數Q，表示小Hi總共詢問的次數與商品的重量被更改的次數之和。

每組測試數據的第N+4~N+Q+3行，每行分別描述一次操作，每行的開頭均爲一個屬於0或1的數字，分別表示該行描述一個詢問和描述一次商品的重量的更改兩種情況。對於第N+i+3行，如果該行描述一個詢問，則接下來爲兩個整數Li, Ri，表示小Hi詢問的一個區間[Li, Ri]；如果該行描述一次商品的重量的更改，則接下來爲兩個整數Pi，Wi，表示位置編號爲Pi的商品的重量變更爲Wi

對於100%的數據，滿足N<=10^6，Q<=10^6, 1<=Li<=Ri<=N，1<=Pi<=N, 0<weight_i, Wi<=10^4。

輸出

對於每組測試數據，對於每個小Hi的詢問，按照在輸入中出現的順序，各輸出一行，表示查詢的結果：標號在區間[Li, Ri]中的所有商品中重量最輕的商品的重量。

樣例輸入

10
3655 5246 8991 5933 7474 7603 6098 6654 2414 884 
6
0 4 9
0 2 10
1 4 7009
0 5 6
1 3 7949
1 3 1227

樣例輸出

2414
884
7474

題目其實很簡單，就一個簡單的線段樹結構，可是寫完之後提交答案總是超時，這是超時的代碼：

#include<iostream>
#include<algorithm>
#define MAX 1000000
using namespace std;
struct NODE
{
	int low, high;
	int value;
}node[4*MAX];
void Init(int num, int low, int high)
{
	if (low == high)
	{
		node[num].high = node[num].low = low;
		cin >> node[num].value;
		return;
	}
	int mid = (low + high) >> 1;
	node[num].high = high;
	node[num].low = low;
	Init(2 * num, low, mid);
	Init(2 * num+1, mid + 1, high);
	node[num].value = min(node[num * 2].value,node[num * 2 + 1].value);
	return;
}
void update(int num, int pos, int revise)
{
	if (node[num].low == node[num].high)
	{
		node[num].value = revise;return;
	}
	int mid = node[num * 2].high;
	if (mid >= pos)
		update(num * 2, pos, revise);
	else
		update(num * 2 + 1, pos, revise);
	node[num].value = min(node[num * 2].value,node[num * 2 + 1].value);
}
int query(int num, int low, int high)
{
	if (node[num].low == low&&node[num].high == high)
		return node[num].value;
	int mid = node[num * 2].high;
	if (mid >= high)
		return query(num * 2, low, high);
	if (mid<low)
		return query(num * 2 + 1, low, high);
	int le = query(num * 2, low, mid);
	int ri = query(num * 2 + 1, mid + 1, high);
	return min(le,ri);
}
int main()
{
	int n;
	cin >> n;
	Init(1, 1, n);
	int m;
	cin >> m;
	for (int i = 0;i<m;i++)
	{
		int a, b, c;
		cin >> a >> b >> c;
		switch (a)
		{
		case 0:cout << query(1, b, c)<<endl;break;
		case 1:update(1, b, c);break;
		}
	}
	return 0;}

最後想到可能是數據量太大，輸入輸出比較多，而C++的cin,cout效率又比較低，所以改成了scanf和printf，結果真的過了，所以在競賽中，輸入輸入較多的話，還是用scanf和printf比較好。

#include<iostream>
#include<cstdio>
#include<algorithm>
#define MAX 1000000
using namespace std;
struct NODE
{
	int low, high;
	int value;
}node[4*MAX];
void Init(int num, int low, int high)
{
	if (low == high)
	{
		node[num].high = node[num].low = low;
		scanf("%d",&node[num].value);
		return;
	}
	int mid = (low + high) >> 1;
	node[num].high = high;
	node[num].low = low;
	Init(2 * num, low, mid);
	Init(2 * num+1, mid + 1, high);
	node[num].value = min(node[num * 2].value,node[num * 2 + 1].value);
	return;
}
void update(int num, int pos, int revise)
{
	if (node[num].low == node[num].high)
	{
		node[num].value = revise;return;
	}
	int mid = node[num * 2].high;
	if (mid >= pos)
		update(num * 2, pos, revise);
	else
		update(num * 2 + 1, pos, revise);
	node[num].value = min(node[num * 2].value,node[num * 2 + 1].value);
}
	
int query(int num, int low, int high)
{
	if (node[num].low == low&&node[num].high == high)
		return node[num].value;
	int mid = node[num * 2].high;
	if (mid >= high)
		return query(num * 2, low, high);
	if (mid<low)
		return query(num * 2 + 1, low, high);
	int le = query(num * 2, low, mid);
	int ri = query(num * 2 + 1, mid + 1, high);
	return min(le,ri);
}
int main()
{
	int n;
	scanf("%d", &n);
	Init(1, 1, n);
	int m;
	scanf("%d", &m);
	for (int i = 0;i<m;i++)
	{
		int a, b, c;
		scanf("%d%d%d", &a, &b, &c);
		switch (a)
		{
		case 0:printf("%d\n", query(1, b, c));break;
		case 1:update(1, b, c);break;
		}
	}
	return 0;
}