HDU 4288 Coder 线段树||暴力

传送门：HDU4288

题意：维护一个队列，三种操作：add  x   将x加入队列（有序插入），del  x   将x从队列删除，  sum  输出队列中下标i%5==3的和。

这题当时在暑假是用线段树写的。。当然是对着题解。。现在对线段树还是一窍不通。。这次又做发现好像暴力也能过，心想这次终于不用看题解了吧。。没想到啊。。暴力怎么改怎么TLE，对着题解还改了好几遍。。最终跟题解写法差不多了才过。。感觉自己写的暴力和题解的没多大差距啊，TLE的不明所以。。

暴力解法：

#include <stdio.h>
#include <string.h>
using namespace std;
int num[100010],n,k;
char s[10];
long long sum;
int main()
{
	int j,a;
	while(~scanf("%d",&n))
	{
		k=0;
		while(n--)
		{
			scanf("%s",s);
			if(!strcmp(s,"add"))
			{
				scanf("%d",&a);
				//printf("%d",a);
				for(j=k++;j>0;j--)
				{
					if(num[j-1]>a)
					num[j]=num[j-1];
					else
					break;
				}
					num[j]=a;
			}
			else if(!strcmp(s,"del"))
			{
				scanf("%d",&a);
				for(j=0;j<k;j++)
					if(num[j]==a)
						break;
				for(;j<k;j++)
				num[j]=num[j+1];
				k--;
			}
			else if(!strcmp(s,"sum"))
			{
				sum=0;
				for(j=2;j<k;j+=5)
				sum+=num[j];
				printf("%lld\n",sum);
			}
		}
	}
return 0;
}

从前的从前写的现在也不明白的线段树：

#include<stdio.h>
#include<algorithm>
#include<string.h> 
#define M 100010
using namespace std;
struct node{
	int l,r,cnt;
	long long sum[5];
}tree[M*3];
char s[M][5];
int len=0,add[M],del[M],flag,num[M];
void build(int l,int r,int root)
{
	tree[root].l=l;
	tree[root].r=r;
	tree[root].cnt=0;
	memset(tree[root].sum,0,sizeof(tree[root].sum));
	if(l!=r)
	{
		int mid=(l+r)/2;
		build(l,mid,2*root);
		build(mid+1,r,2*root+1);
	}

}
int find(int k)//二分查找k应该是第几个数 
{
	int l=0,r=len-1;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(add[mid]<k)
		l=mid+1;
		else if(add[mid]>k)
		r=mid-1;
		else
		return mid;
	}
}
void update(int k)//离散化关键步骤  弄明白 （原来写的，到现在也不明白为啥。。）
{
	for(int i=0;i<5;i++)
	tree[k].sum[i]=tree[2*k].sum[i]+tree[2*k+1].sum[((i-tree[2*k].cnt)%5+5)%5];
	//这个区间的和是左右子树同样余数的相加和，
	//右子树要用序号减去左子树的个数对5取模后为了保持正数所以要+5再取模，所以是(i-a[x*2].cnt)%5+5  
}
int insert(int root,int t,int pos)
{
	tree[root].cnt+=2*flag-1;//区间内有几个数字
	if(tree[root].l==tree[root].r)
	{
		tree[root].sum[0]=flag*t;//巧用flag实现若del则清零 若add则赋值  
		return 1;
	}
	int mid=(tree[root].l+tree[root].r)/2;
	if(pos<=mid)
	insert(2*root,t,pos);
	else 
	insert(2*root+1,t,pos);
	update(root);
}
int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		getchar();
		int j=0;
		for(int i=0;i<n;i++)
		{
			scanf("%s",s[i]);
			if(s[i][0]!='s')
			scanf("%d",&num[i]);
			if(s[i][0]=='a')
			add[len++]=num[i];
		}
		sort(add,add+len);
		build(1,n,1);
		for(int i=0;i<n;i++)
		{
			if(s[i][0]=='a')
			{
				flag=1;
				int pos=find(num[i]);
				insert(1,num[i],pos);
			}
			else if(s[i][0]=='d')
			{
				flag=0;
				int pos=find(num[i]);
				insert(1,num[i],pos);
			}
			else
			printf("%lld\n",tree[1].sum[2]);
		}
		len=0;
	}
	
}

17.2.4记：

前两天下决心要搞明白线段树，开始刷线段树的题，再次碰到这个题，也是我第三次碰到这个题了，虽然还得重温一遍题解，但这次我真正弄明白了update部分为什么要那么写，参考：http://blog.csdn.net/dgq8211/article/details/7999179     感觉这种写法更好理解一些吧，两种写法的区别不过是一个根据左区间下标推右区间下标，另一个反过来而已，本质是一样的。核心部分明白了，也算对得起三刷这道题吧。