POJ3468

最新推荐文章于 2018-02-28 16:06:38 发布
最新推荐文章于 2018-02-28 16:06:38 发布
阅读量857 收藏 2
点赞数
分类专栏： 数据结构_伸展树 文章标签： POJ3468
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/scut_pein/article/details/18950277
版权 
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define maxn 100010
#define inf 0x3f3f3f3f
#define LL long long int
#define Key_value ch[ch[root][1]][0]
int pre[maxn],key[maxn],ch[maxn][2],root,tot1;
int size[maxn],s[maxn],tot2,val[maxn];
int add[maxn];
LL sum[maxn];
int n,q;
void NewNode(int & r,int father,int k)
{
	if(tot2)
		r = s[--tot2];
	else 
		r = ++tot1;
	pre[r] = father;
	val[r] = k;
	sum[r] = k;
	add[r] = 0;
	size[r] = 1;
	ch[r][0] = ch[r][1] = 0;
}

//将延迟标记更新到孩子结点
void PushDown(int r)
{
	if(add[r])
	{
		val[r] += add[r];
		add[ch[r][0]] += add[r];
		add[ch[r][1]] += add[r];
		sum[ch[r][0]] += (LL)add[r]*size[ch[r][0]];
		sum[ch[r][1]] += (LL)add[r]*size[ch[r][1]];
		add[r] = 0;
	}
}

//通过孩子结点更新父结点
void PushUp(int r)
{
	size[r] = size[ch[r][0]] + size[ch[r][1]] + 1;
	sum[r] = sum[ch[r][0]] + sum[ch[r][1]] + val[r] + add[r];
}

//旋转，kind为1为右旋，kind为0为左旋
void Rotate(int x,int kind)
{
	int y = pre[x];
	PushDown(y);
	PushDown(x);
	//类似SBT,要把其中一个分支先给父结点
	ch[y][!kind] = ch[x][kind];
	pre[ch[x][kind]] = y;
	//如果父结点不是根结点，则要和父结点的父结点连接起来
	if(pre[y])
		ch[pre[y]][ch[pre[y]][1] == y] = x;
	pre[x] = pre[y];
	ch[x][kind] = y;
	pre[y] = x;
	PushUp(y);
	//PushUp(x);
}

void Splay(int r,int goal)
{
	PushDown(r);
	while(pre[r] != goal)
	{
		//父结点即是目标位置，goal为0表示，父结点就是根节点
		if(pre[pre[r]] == goal)
			Rotate(r,ch[pre[r]][0]==r);
		else
		{
			int y = pre[r];
			int kind = ch[pre[y]][0] == y;
			//两个方向不同，则先左旋再右旋
			if(ch[y][kind] == r)
			{
				Rotate(r,!kind);
				Rotate(r,kind);
			}
			//两个方向相同，相同方向连续两次
			else 
			{
				Rotate(y,kind);
				Rotate(r,kind);
			}
		}
	}
	PushUp(r);
	//更新根节点
	if(goal == 0)	root = r;
}

//把第k位的数转到goal下边
void RotateTo(int k,int goal)
{
	int r = root;
	PushDown(r);
	while(size[ch[r][0]] != k)
	{
		if(k < size[ch[r][0]])
			r = ch[r][0];
		else 
		{
			k -= (size[ch[r][0]] + 1);
			r = ch[r][1];
		}
		PushDown(r);
	}
	Splay(r,goal);
}

int Insert(int k)
{
	int r = root;
	while(ch[r][key[r]<k])
		r = ch[r][key[r] < k];
	NewNode(ch[r][k>key[r]],r,k);
	Splay(ch[r][k>key[r]],0);
	return 1;
}

//找前驱，即左子树的最右结点
int get_pre(int x)
{
	int tmp = ch[x][0];
	if(tmp == 0)	return inf;
	while(ch[tmp][1])
		tmp = ch[tmp][1];
	return key[x] - key[tmp];
}

//找后继，即右子树的最左结点
int get_next(int x)
{
	int tmp = ch[x][1];
	if(tmp == 0)	return inf;
	while(ch[tmp][0])
		tmp = ch[tmp][0];
	return key[tmp] - key[x];
}

//查询[l,r]之间的和
LL Query(int l,int r)
{
	RotateTo(l-1,0);
	RotateTo(r+1,root);
	return sum [Key_value];
}

//更新
void Update(int l,int r)
{
	int k;
	scanf("%d",&k);
	RotateTo(l-1,0);
	RotateTo(r+1,root);
	add[Key_value] += k;
	sum[Key_value] += size[Key_value]*k;
}

int a[maxn];
//建树，中间结点先建立，然后分别对区间两端在左右子树建立
void BuildTree(int &x,int l,int r,int father)
{
	if(l > r)
		return;
	int mid = (l+r)/2;
	NewNode(x,father,a[mid]);
	if(l < mid)
		BuildTree(ch[x][0],l,mid-1,x);
	if(r > mid)
		BuildTree(ch[x][1],mid+1,r,x);
	PushUp(x);
}

void Init()
{
	for(int i = 0;i < n;i++)
		scanf("%d",&a[i]);
	ch[0][0] = ch[0][1] = pre[0] = size[0] = 0;
	add[0] = sum[0] = 0;
	root = tot1 = 0;
	NewNode(root,0,-1);
	NewNode(ch[root][1],root,-1);
	size[root] = 2;
	BuildTree(Key_value,0,n-1,ch[root][1]);//让所有数据夹在两个-1之间
	PushUp(ch[root][1]);
	PushUp(root);
}

int main()
{
	while(scanf("%d%d",&n,&q)!=EOF)
	{
		Init();
		while(q--)
		{
			char str[10];
			int x,y;
			scanf("%s%d%d",str,&x,&y);
			if(str[0] == 'Q')
				printf("%lld\n",Query(x,y));
			else 
				Update(x,y);
		}
	}
	return 0;
}



风格更新后：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define inf 0x3f3f3f3f
#define maxn 100010
#define LL long long int
#define Key_value ch[ch[root][1]][0]

inline int min(int a,int b)
{
	return a>b?b:a;
}

struct SplayTree
{
	int pre[maxn],key[maxn],ch[maxn][2],size[maxn],add[maxn],a[maxn],root,cnt;
	LL sum[maxn];
	void init(int n)
	{
		root = cnt = 0;
		ch[0][0] = ch[0][1] = pre[0] = size[0] = key[0] = 0;
		add[0] = sum[0] = 0;
		NewNode(root,0,-1);
		NewNode(ch[root][1],root,-1);
		size[root] = 2;
		BuildTree(Key_value,0,n-1,ch[root][1]);
		PushUp(ch[root][1]);
		PushUp(root);
	}

	void BuildTree(int & x,int l,int r,int father)
	{
		if(l > r)	return;
		int mid = (l+r) >> 1;
		NewNode(x,father,a[mid]);
		if(l < mid)
			BuildTree(ch[x][0],l,mid-1,x);
		if(r > mid)
			BuildTree(ch[x][1],mid+1,r,x);
		PushUp(x);
	}
	
	void NewNode(int & r,int father,int k)
	{
		r = ++cnt;
		pre[r] = father;
		ch[r][0] = ch[r][1] = 0;
		key[r] = sum[r] = k;
		add[r] = 0;
		size[r] = 1;
	}

	void PushDown(int r)
	{
		if(add[r])
		{
			key[r] += add[r];
			add[ch[r][0]] += add[r];
			add[ch[r][1]] += add[r];
			sum[ch[r][0]] += (LL)add[r]*size[ch[r][0]];
			sum[ch[r][1]] += (LL)add[r]*size[ch[r][1]];
			add[r] = 0;
		}
	}

	void PushUp(int r)
	{
		size[r] = size[ch[r][0]] + size[ch[r][1]] + 1;
		sum[r] = sum[ch[r][0]] + sum[ch[r][1]] + key[r] + add[r];
	}

	void Rotate(int x,int kind)
	{
		int y = pre[x];
		PushDown(y);
		PushDown(x);
		ch[y][!kind] = ch[x][kind];
		pre[ch[x][kind]] = y;
		if(pre[y])
			ch[pre[y]][ch[pre[y]][1]==y] = x;
		pre[x] = pre[y];
		ch[x][kind] = y;
		pre[y] = x;
		PushUp(y);
	}
	
	void Splay(int r,int goal)
	{
		while(pre[r] != goal)
		{
			int y = pre[r],z = pre[y];
			PushDown(z);	PushDown(y);	PushDown(r);
			if(pre[pre[r]] == goal)
			{
				Rotate(r,ch[pre[r]][0]==r);
			}
			else 
			{
				int y = pre[r];
				int kind = ch[pre[y]][0] == y;
				if(ch[y][kind] == r)
				{
					Rotate(r,!kind);
				}
				else
				{
					Rotate(y,kind);
					Rotate(r,kind);
				}
			}
		}
		PushUp(r);
		if(goal == 0)	root = r;
	}

	void RotateTo(int k,int goal)
	{
		int r = root;
		PushDown(r);
		while(size[ch[r][0]] != k)
		{
			if(k < size[ch[r][0]])
				r = ch[r][0];
			else
			{
				k -= (size[ch[r][0]] + 1);
				r = ch[r][1];
			}
			PushDown(r);
		}
		Splay(r,goal);
	}

	void Update(int l,int r)
	{
		int k;
		scanf("%d",&k);
		RotateTo(l-1,0);
		RotateTo(r+1,root);
		add[Key_value] += k;
		sum[Key_value] += size[Key_value]*k;
	}

	LL Query(int l,int r)
	{
		RotateTo(l-1,0);
		RotateTo(r+1,root);
		return sum[Key_value];
	}

	void solve(int n,int q)
	{
		for(int i = 0;i < n;i++)
		{
			scanf("%d",&a[i]);
		}
		init(n);
		while(q--)
		{
			char str[10];
			int x,y;
			scanf("%s%d%d",str,&x,&y);
			if(str[0] == 'Q')
				printf("%lld\n",Query(x,y));
			else Update(x,y);
		}
	}
}spt;

int main()
{
	int n,q;
	while(scanf("%d%d",&n,&q)!=EOF)
	{
		spt.solve(n,q);
	}
	return 0;
}


SCUT_Pein
关注
  • 0
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
线段树入门学习(二）(兼解POJ3468) JAVA
04-14
在本教程中，我们将深入探讨线段树的基本概念，以及如何使用Java实现线段树来解决实际问题，如POJ3468题目。 首先，理解线段树的核心在于它的分治思想。线段树将一个大区间（通常是一维数组）分成多个小区间，每个...
poj.rar_poj
09-23
标题中的"poj.rar_poj"暗示了这是一个与POJ（Programming Online Judge）相关的压缩文件，POJ是一个在线编程竞赛平台，主要供程序员们练习和提交算法解决方案。在这个压缩包中，很可能包含了用户在POJ上参与编程挑战...
poj 3468
weixin_30480651的博客
01-08 58
线段树区间修改入门题目。快被这个长整形给整死啦。以后如果对自己的算法的正确性有80%的把握，就去看一下细节，非常认真的看一遍，一行一行的看，这样其实更省时间，就比如我这次调来调去，调了4个多小时，最后才突然发现。 //线.3段树区间修改 #include <iostream> #include <stdio.h> #include <cstring> usin...
POJ 3468
我不吃海鲜的博客
02-28 218
题目：http://poj.org/problem?id=3468超时代码：#include&lt;bits/stdc++.h&gt; #define INF 1e18 #define inf 1e9 #define min(a,b) a&lt;b?a:b #define max(a,b) a&gt;b?a:b #define lson l,m,rt&lt;&lt;1 #define rson m...
POJ-3468
xhldtc的专栏
09-06 509
成段更新的线段树，应该也算比较基础的写法
POJ3468.zip_poj3468
09-14
标题中的"POJ3468.zip_poj3468"显然与编程竞赛相关，POJ（Problemset Online Judge）是北京大学主办的一个在线编程竞赛平台，其中的3468是一个具体的题目编号。这个题目可能涉及到编程挑战，要求参赛者解决特定的...
POJ3468.zip_ACM
最新发布
09-23
标题中的"POJ3468.zip_ACM"暗示了这是一个与ACM（国际大学生程序设计竞赛）相关的项目。在ACM比赛中，参赛者需要解决各种算法问题，以提高编程和解决问题的能力。"POJ3468"是这个特定问题的编号，在编程竞赛平台上的...
POJ.rar_poj
09-19
【标题】"POJ.rar_poj" 指的是一份与编程竞赛相关的压缩文件，主要涉及POJ（Problem Overloading Judge）平台上的问题解答。POJ是一个在线的编程竞赛平台，它为参赛者提供了大量的算法题目进行练习和比赛，以提升...
poj-3468
fyfcauc的专栏
08-13 431
// 6884K 2032MS G++ #include "cstdio" #include "cstring" #include "iostream" using namespace std; #define MAX 100010 struct TreeNode { long left; long right; long long sum; long
poj3468
active2489595970的博客
10-05 309
You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for th
POJ2761--Feed the dogs
SCUT_Pein
11-14 1077
Description Wind loves pretty dogs very much, and she has n pet dogs. So Jiajia has to feed the dogs every day for Wind. Jiajia loves Wind, but not the dogs, so Jiajia use a special way to feed the d
UVa 11996 Jewel Magic 伸展树
SCUT_Pein
02-27 1063
I am a magician. I have a string of emeralds and pearls. I may insert new jewels in the string, or remove old ones. I may even reverse a consecutive part of the string. At anytime, if you point to two
UVA 11922 Permutation Transformer 伸展树
SCUT_Pein
02-25 846
Write a program to transform the permutation 1, 2, 3,..., n according to m instructions. Each instruction(a, b) means to take out the subsequence from the a-th to the b-th element, reverse it, then
HDU1890
SCUT_Pein
02-07 841
#include #include #include using namespace std; #define Keytree (ch[ ch[root][1] ][0]) const int maxn = 100010; struct node { int num,id; }in[maxn]; int id[maxn]; int cmp(node a,node b){
HDU3436
SCUT_Pein
02-07 778
#include #include #include #include #define N 100015 #define inf 1<<29 #define LL long long #define Key_value ch[ch[root][1]][0] using namespace std; int n,q,p[N],cnt,s[2*N],e[2*N],ope[N]; i
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值