1. 程式人生 > >【Vijos1448】校門外的樹

【Vijos1448】校門外的樹

                                           校門外的樹

描述

校門外有很多樹,有蘋果樹,香蕉樹,有會扔石頭的,有可以吃掉補充體力的……
如今學校決定在某個時刻在某一段種上一種樹,保證任一時刻不會出現兩段相同種類的樹,現有兩個操作:
K=1,K=1,讀入l、r表示在區間[l,r]中種上一種樹,每次操作種的樹的種類都不同
K=2,讀入l,r表示詢問l~r之間能見到多少種樹
(l,r>0)

格式

輸入格式

第一行n,m表示道路總長為n,共有m個操作
接下來m行為m個操作

輸出格式

對於每個k=2輸出一個答案

樣例1

樣例輸入1

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

樣例輸出1

1
2

限制

1s

提示

範圍:20%的資料保證,n,m<=100
60%的資料保證,n <=1000,m<=50000
100%的資料保證,n,m<=50000

解析:

       一道挺有意思的題。對於每個區間(L , R),我們可以理解為在 L 處加入一個 '(' ,在 R 處加入一個 ')' ,答案就為 R 之前的 ‘(’ 的數量減去 L-1 之前的 ‘)’ 的數量,用樹狀陣列可以輕鬆解決。

程式碼:

#include <bits/stdc++.h>
using namespace std;

const int Max=50010;
int n,m;
int sum1[Max],sum2[Max];

inline int get_int()
{
	int x=0,f=1;
	char c;
	for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
	if(c=='-') {f=-1;c=getchar();}
	for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
	return x*f;
}

inline int lowbit(int x){return x&-x;}
inline void add1(int pos,int x){for(;pos<=n;pos+=lowbit(pos)) sum1[pos]+=x;}
inline void add2(int pos,int x){for(;pos<=n;pos+=lowbit(pos)) sum2[pos]+=x;}
inline int Q1(int pos)
{
	int ans=0;
	for(;pos;pos-=lowbit(pos)) ans+=sum1[pos];
	return ans;
}
inline int Q2(int pos)
{
	int ans=0;
	for(;pos;pos-=lowbit(pos)) ans+=sum2[pos];
	return ans;
}

int main()
{
	n=get_int(),m=get_int();
	for(int i=1;i<=m;i++)
	{
	  int flag=get_int(),l=get_int(),r=get_int();
	  if(flag == 1) add1(l,1),add2(r,1);
	  else cout<<Q1(r) - Q2(l-1)<<"\n";
	}

	return 0;
}