程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> 關於C++ >> bzoj3110【ZJOI2013】K大數查詢

bzoj3110【ZJOI2013】K大數查詢

編輯:關於C++

Description

有N個位置,M個操作。操作有兩種,每次操作如果是1 a b c的形式表示在第a個位置到第b個位置,每個位置加入一個數c
如果是2 a b c形式,表示詢問從第a個位置到第b個位置,第C大的數是多少。

Input

第一行N,M
接下來M行,每行形如1 a b c或2 a b c

Output

輸出每個詢問的結果

Sample Input

2 5
1 1 2 1
1 1 2 2
2 1 1 2
2 1 1 1
2 1 2 3

Sample Output

1
2
1

HINT

【樣例說明】

第一個操作 後位置 1 的數只有 1 , 位置 2 的數也只有 1 。 第二個操作 後位置 1

的數有 1 、 2 ,位置 2 的數也有 1 、 2 。 第三次詢問 位置 1 到位置 1 第 2 大的數 是

1 。 第四次詢問 位置 1 到位置 1 第 1 大的數是 2 。 第五次詢問 位置 1 到位置 2 第 3

大的數是 1 。?


N,M<=50000,N,M<=50000

a<=b<=N

1操作中abs(c)<=N

2操作中abs(c)<=Maxlongint

Source

線段樹套線段樹+二分答案

樹套樹竟然一次就寫對了,那叫一個開心啊!

外層是權值線段樹,記錄權值為[l,r]的信息。

內層是區間線段樹,記錄當前權值的位置在[l,r]的數量。

每次加點,把外層所有覆蓋該權值的線段樹全部更新。

每次查詢,二分答案即可。

#include
#include
#include
#include
#include
#include
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 20000005
using namespace std;
int n,m,f,a,b,c,tot;
int rt[maxn],ls[maxn],rs[maxn],sum[maxn],tag[maxn];
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void update(int k,int l,int r,int x)
{
	tag[k]+=x;
	sum[k]+=(r-l+1)*x;
}
inline void pushdown(int k,int l,int r)
{
	if (!tag[k]||l==r) return;
	if (!ls[k]) ls[k]=++tot;
	if (!rs[k]) rs[k]=++tot;
	int mid=(l+r)>>1;
	update(ls[k],l,mid,tag[k]);
	update(rs[k],mid+1,r,tag[k]);
	tag[k]=0;
}
inline void pushup(int k)
{
	sum[k]=sum[ls[k]]+sum[rs[k]];
}
inline void add(int &k,int l,int r,int a,int b)
{
	if (!k) k=++tot;
	if (l==a&&r==b)
	{
		update(k,l,r,1);
		return;
	}
	pushdown(k,l,r);
	int mid=(l+r)>>1;
	if (b<=mid) add(ls[k],l,mid,a,b);
	else if (a>mid) add(rs[k],mid+1,r,a,b);
	else add(ls[k],l,mid,a,mid),add(rs[k],mid+1,r,mid+1,b);
	pushup(k);
}
inline void insert()
{
	int k=1,l=1,r=n;
	while (l!=r)
	{
		int mid=(l+r)>>1;
		add(rt[k],1,n,a,b);
		if (c<=mid) r=mid,k=k<<1;
		else l=mid+1,k=k<<1|1;
	}
	add(rt[k],1,n,a,b);
}
inline int query(int k,int l,int r,int a,int b)
{
	if (!k) return 0;
	if (l==a&&r==b) return sum[k];
	pushdown(k,l,r);
	int mid=(l+r)>>1;
	if (b<=mid) return query(ls[k],l,mid,a,b);
	else if (a>mid) return query(rs[k],mid+1,r,a,b);
	else return query(ls[k],l,mid,a,mid)+query(rs[k],mid+1,r,mid+1,b);
}
inline int solve()
{
	int k=1,l=1,r=n;
	while (l!=r)
	{
		int mid=(l+r)>>1;
		int tmp=query(rt[k<<1|1],1,n,a,b);
		if (tmp>=c) l=mid+1,k=k<<1|1;
		else c-=tmp,r=mid,k=k<<1;
	}
	return l;
}
int main()
{
	n=read();m=read();
	while (m--)
	{
		f=read();a=read();b=read();c=read();
		if (f==1) insert();
		else printf("%d\n",solve());
	}
}
  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved