POJ 2823 Sliding Window（线段树入门）

题意:

8 3
1 3 -1 -3 5 3 6 7

一串数列，有一个窗口大小为3，从数列开始往后移动，输出最大和最小值。

-1 -3 -3 -3 3 3
3 3 5 5 6 7

窗口大小为3

思路：

维护一个线段树，代码很详细

解题心得：

因为关键值的输入量有1000000，也就是叶节点有1000000个，总节点按理说是2000000-1，但这题得开3000000才能过

代码：

#include<stdio.h>
int max(int a,int b)
{
	return a>b?a:b;
}
int min(int a,int b)
{
	return a<b?a:b;
}



struct Node
{
	int L,R;
	int maxn,minn;
};
Node Tree[3000000];//线段树
int a[1000001];//存输入数据
int ans_max[1000001],ans_min[1000001];//存输出数据



//建树，节点为Tree[root],线段范围为l到r
void Maketree(int root,int l,int r)
{
	//1.给线段范围赋值
	Tree[root].L=l;
	Tree[root].R=r;
	//2.判断是否已经到达叶节点，若是则赋关键值并返回，否则继续延伸
	if(l==r)
	{
		Tree[root].maxn=Tree[root].minn=a[l];//关键值即为题目要维护的元素。曾错在此处
		return;
	}
	else
	{
		//取要分割线段的中点
		int mid=(l+r)>>1;
		//建立左儿子
		Maketree(root*2,l,mid);
		//建立右儿子
		Maketree(root*2+1,mid+1,r);
		//根据左儿子与右儿子的关键值，更新自身关键值
		Tree[root].maxn=max(Tree[root*2].maxn,Tree[root*2+1].maxn);
		Tree[root].minn=min(Tree[root*2].minn,Tree[root*2+1].minn);
		return;
	}
}


//寻找以root为根中，范围为l到r的线段的关键值，结果存在maxn与minn中
void find(int root,int l,int r,int& maxn,int& minn)
{
	//如果树根的线段正好满足，存结果并返回,否则往下搜索
	if(Tree[root].L==l&&Tree[root].R==r)
	{
		maxn=Tree[root].maxn;
		minn=Tree[root].minn;
		return;
	}
	else
	{
		//取该树根线段的中点
		int mid=(Tree[root].L+Tree[root].R)>>1;
		//如果要查找的线段在树根线段中点左侧 T->L---l----r----mid--------T->R
		if(mid>=r)
		{
			//往左儿子寻找
			find(root*2,l,r,maxn,minn);
		}
		//如果要查找的线段在树根中点右侧
		else if(mid<l)
		{
			//往右儿子寻找
			find(root*2+1,l,r,maxn,minn);
		}
		//如果要查找的线段在树根的左儿子和右儿子中各占一部分
		else
		{
			//将要查找的线段分为两部分，一部分往左儿子找，一部分往右儿子找
			int max1,max2,min1,min2;//存放两边各自找到的关键值，取最优
			//左儿子部分
			find(root*2,l,mid,max1,min1);
			//右儿子部分
			find(root*2+1,mid+1,r,max2,min2);
			//取最优更新在此树根找到的结果
			maxn=max(max1,max2);
			minn=min(min1,min2);
		}
		return;
	}
}

int main()
{
	int n,k;
	while(~scanf("%d%d",&n,&k))
	{
		int i;
		for(i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
		}
		Maketree(1,1,n);
		for(i=1;i<=n-k+1;i++)
		{
			find(1,i,i+k-1,ans_max[i],ans_min[i]);
		}
		for(i=1;i<=n-k+1;i++)
		{
			if(i==1) printf("%d",ans_min[i]);
			else printf(" %d",ans_min[i]);
		}
		printf("\n");
		for(i=1;i<=n-k+1;i++)
		{
			if(i==1) printf("%d",ans_max[i]);
			else printf(" %d",ans_max[i]);
		}
		printf("\n");
	}
	return 0;
}