【线段树】 HDOJ 4031 Attack

个人认为这题很不咋样。。。用的算法最坏情况下是n方。。。

#include <iostream>  
#include <queue>  
#include <stack>  
#include <map>  
#include <set>  
#include <bitset>  
#include <cstdio>  
#include <algorithm>  
#include <cstring>  
#include <climits>  
#include <cstdlib>
#include <cmath>
#include <time.h>
#define maxn 20005
#define maxm 400005
#define eps 1e-10
#define mod 3
#define INF 1e17
#define lowbit(x) (x&(-x))  
#define ls o<<1
#define rs o<<1 | 1
#define lson o<<1, L, mid  
#define rson o<<1 | 1, mid+1, R  
typedef long long LL;
//typedef int LL;
using namespace std;

int sum[maxn<<2];
int mark[maxn<<2];
struct node
{
	int a, b;
}att[maxn];
int n, m, ql, qr, p, t;

void init(void)
{
	memset(sum, 0, sizeof sum);
	memset(mark, 0, sizeof mark);
}
void pushdown(int o)
{
	if(!mark[o]) return;
	sum[ls] += mark[o];
	sum[rs] += mark[o];
	mark[ls] += mark[o];
	mark[rs] += mark[o];
	mark[o] = 0;
}
void pushup(int o)
{
	sum[o] = sum[ls] + sum[rs];
}
void updata(int o, int L, int R)
{
	if(ql <= L && qr >= R) {
		sum[o] += 1;
		mark[o] += 1;
		return;
	}
	pushdown(o);
	int mid = (L+R)>>1;
	if(ql <= mid) updata(lson);
	if(qr > mid) updata(rson);
	pushup(o);
}
int  query(int o, int L, int R)
{
	if(L == R) return sum[o];
	pushdown(o);
	int mid = (L+R)>>1, ans = 0;
	if(p <= mid) ans = query(lson);
	else ans = query(rson);
	pushup(o);
	return ans;
}
void work(void)
{
	int cnt = 0;
	char s[20];
	while(m--) {
		scanf("%s", s);
		if(s[0] == 'A') {
			scanf("%d%d", &ql, &qr);
			att[cnt].a = ql, att[cnt++].b = qr;
			updata(1, 1, n);
		}
		else {
			int ans = 0;
			scanf("%d", &p);
			for(int i = 0; i < cnt; i++) {
				if(att[i].a <= p && att[i].b >= p) i += t-1, ans++;
			}
			printf("%d\n", query(1, 1, n) - ans);
		}
	}
}
int main(void)
{
	int _, __ = 0;
	while(scanf("%d", &_)!=EOF) {
		while(_--) {
			init();
			scanf("%d%d%d", &n, &m, &t);
			printf("Case %d:\n", ++__);
			work();
		}
	}
	return 0;
}