【2015ZUFE新生赛网络同步赛L】【线段树找最小值最早出现位置 破环成链】God Wang 环上找小于等于tim的第一个数

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<ctype.h>
#include<math.h>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<bitset>
#include<algorithm>
#include<time.h>
using namespace std;
void fre(){freopen("c://test//input.in","r",stdin);freopen("c://test//output.out","w",stdout);}
#define MS(x,y) memset(x,y,sizeof(x))
#define MC(x,y) memcpy(x,y,sizeof(x))
#define MP(x,y) make_pair(x,y)
#define ls o<<1
#define rs o<<1|1
typedef long long LL;
typedef unsigned long long UL;
typedef unsigned int UI;
template <class T1,class T2>inline void gmax(T1 &a,T2 b){if(b>a)a=b;}
template <class T1,class T2>inline void gmin(T1 &a,T2 b){if(b<a)a=b;}
const int N=1e5+10,M=0,Z=1e9+7,ms63=1061109567;
int casenum,casei;
int n,K;
int v[N<<1];
int sum[N];
struct A
{
	int l,r;
	int minv;
}a[1<<19];
int ans[N];
inline void pushup(int o)
{
	a[o].minv=min(a[ls].minv,a[rs].minv);
}
void build(int o,int l,int r)
{
	a[o].l=l;
	a[o].r=r;
	if(l==r)
	{
		a[o].minv=v[l];
		return;
	}
	int m=(l+r)>>1;
	build(ls,l,m);
	build(rs,m+1,r);
	pushup(o);
}
int tim,Y,P;
void find(int o)
{
	if(a[o].minv>tim)return;
	if(a[o].l==a[o].r)
	{
		P=a[o].l;
		a[o].minv=1e9;
		return;
	}
	if(a[ls].r>=Y)find(ls);
	if(P==0)find(rs);
	pushup(o);
}
void change(int o)
{
	if(a[o].l==a[o].r)
	{
		a[o].minv=1e9;
		return;
	}
	int m=(a[o].l+a[o].r)>>1;
	if(P<=m)change(ls);
	else change(rs);
	pushup(o);
}
int main()
{
	while(~scanf("%d%d",&n,&K))
	{
		memset(sum,0,n+2<<2);
		for(int i=1;i<=n;++i)
		{
			scanf("%d",&v[i]);v[i]-=K;
			++sum[max(v[i],0)];
		}
		bool flag=1;
		for(int i=1;i<=n;++i)
		{
			if(sum[i-1]<i){flag=0;break;}
			else sum[i]+=sum[i-1];
		}
		if(!flag){puts("-1");continue;}
		for(int i=1;i<=n;++i)v[n+i]=v[i];
		build(1,1,2*n);
		Y=1;for(tim=0;tim<n;++tim)
		{
			P=0;find(1);
			if(P<=n)
			{
				Y=ans[tim+1]=P;
				P+=n;
				change(1);
			}
			else
			{
				Y=ans[tim+1]=P-n;
				P-=n;
				change(1);
			}
		}
		for(int i=1;i<n;++i)printf("%d ",ans[i]);
		printf("%d\n",ans[n]);
	}
	return 0;
}
/*
【题意】
有n（1<=n<=1e5）颗糖果，形成一个环。
我们当前在位置1，按照1->2->3->...->n->1的顺序不停走。
初始时刻，我们的体重是k，每吃一颗糖果，体重增加1。
每个点上有一颗糖果，每个糖果有属性a[]，表示要吃这颗糖果，我们的体重至少为a[]
让你判定是否可以最终吃完全部的糖果。
如果无法吃完，输出-1。否则按照我们吃糖果的顺序，输出所有糖果的编号。

【类型】
线段树

【分析】
我们发现，我们本质上，其实只是找在一个位置后的最近的一个<=一定值的数的位置。
然后我们取走这个数值，并重复操作，直到最后取走所有的数。
转换一下，如何找到某个位置后<=一定值的数的位置呢？显然我们可以使用线段树。
线段树记录区间最小值，然后通过类似于二分的做法找到第一个合法位置。

然而还有一个问题，这是一个环，怎么办，size*2，破坏成链。
如果数被取走，则把值修改为极大值1e9，并且也修改它的另外一个映射点。
这样这道题就可以以O(nlogn)的时间AC掉啦！

【时间复杂度&&优化】
O（nlogn）

1，这题，我们可以先把每个a[]-=K，初始体重设为0
这样，当我们要第i次吃糖果的时候，我们的体重为i-1，这颗糖果的使用要求就必须是<=i-1
我们可以求一个前缀和，sum[i-1]至少要为i，以此来起到程序加速的作用。

2，线段树我们发现，我们是找每个节点（设为Y）往右的第一个权值<=tim的糖果的下标，只要保证左界>=Y即可。右界无要求。
所以我们可以对线段树做适当修改。
从根节点开始走，如果区间最小值比tim大，则必然无法更新。
如果左区间有超过Y的部分，先尝试在左区间找答案，否则在右区间找答案。

【数据】
5 0
4 3 2 1 0

*/