Codeforces.449C.Willem, Chtholly and Seniorious(ODT)

题目链接


(Description)
给定一个序列,随机四种操作:
1.区间加x。
2.区间覆盖为x。
3.输出区间第k大的数。
4.输出区间k次幂的和。

(Solution)
ODT,利用set维护同色连续区间的暴力算法。
真的就是纯暴力。。随机数据和区间赋值很多的题跑过是没大问题的。
非数据随机随便被卡。一般不考虑。不过可以写个板子。

要注意的:

判断it是否为空
序列中的数是longlong,快速幂前要取模。
如果先L=Split(l)的话Split(r+1)时可能将L这个迭代器删掉导致RE
set.insert返回值:pair<set::iterator, bool> (iterator:插入位置 bool:是否成功插入不可重集)
set.erase(beg,end):删除迭代器区间[beg,end)中的所有元素,返回下一个元素的迭代器
对于要用迭代器修改的内容,mutable是必要的


//343ms	5500KB
#include <set>
#include <cstdio>
#include <algorithm>
#define mp std::make_pair
#define pr std::pair<LL,int>
#define Iter std::set<Node>::iterator
typedef long long LL;
const int N=1e5+5;

LL seed;
struct Node
{
	int l,r;
	mutable LL v;
	Node(int l=0,int r=-1,LL v=0):l(l),r(r),v(v) {}
	bool operator <(const Node &x)const
	{
		return l<x.l;
	}
};
std::set<Node> st;

inline int Rand()
{
	int tmp=seed; seed=(seed*7+13)%1000000007;
	return tmp;
}
inline int FP(LL x,int k,int mod)
{
	LL res=1;
	for(; k; k>>=1,x=x*x%mod)
		k&1&&(res=res*x%mod);
	return res;
}
Iter Split(int p)
{
	Iter it=st.lower_bound(Node(p));//Node(p,p,0)
	if(it!=st.end() && it->l==p) return it;//判断it是否为空!
	--it;
	int l=it->l,r=it->r; LL v=it->v;
	st.erase(it), st.insert(Node(l,p-1,v));
	return st.insert(Node(p,r,v)).first;
	//set.insert返回值:pair<set<int>::iterator, bool> (iterator:插入位置 bool:是否成功插入不可重集)
}
LL Kth(int l,int r,int k)
{
	static pr A[N];
	Iter R=Split(r+1),L=Split(l);
	int cnt=0;
	for(; L!=R; ++L) A[++cnt]=mp(L->v,L->r-L->l+1);//longlong!
	std::sort(A+1,A+1+cnt);
	for(int i=1; i<=cnt; ++i)
		if((k-=A[i].second)<=0) return A[i].first;
	return 0;
}
int Sum(int l,int r,int x,int mod)
{
	LL res=0;
	Iter R=Split(r+1),L=Split(l);
	for(; L!=R; ++L) res+=1ll*(L->r-L->l+1)*FP(L->v%mod,x,mod)%mod;//L->v(longlong)要取模!!
	return res%mod;
}

int main()
{
	int n,m,vmax;
	scanf("%d%d%lld%d",&n,&m,&seed,&vmax);
	for(int i=1; i<=n; ++i) st.insert(Node(i,i,Rand()%vmax+1));
	for(int i=1; i<=m; ++i)
	{
		int opt=Rand()%4+1,l=Rand()%n+1,r=Rand()%n+1;
		l>r&&(std::swap(l,r),0);
		switch(opt)
		{
			case 1:
			{
				int x=Rand()%vmax+1;
				Iter R=Split(r+1),L=Split(l);
				for(; L!=R; L->v+=x, ++L);
				break;
			}
			case 2:
			{
				int x=Rand()%vmax+1;
				Iter R=Split(r+1),L=Split(l);
				//如果先L=Split(l)的话Split(r+1)时可能将L这个迭代器删掉导致RE 
				st.erase(L,R), st.insert(Node(l,r,x));
				//set.erase(beg,end):删除迭代器区间[beg,end)中的所有元素,返回下一个元素的迭代器	 
				break;
			}
			case 3: printf("%lld
",Kth(l,r,Rand()%(r-l+1)+1)); break;
			case 4:
			{
				int x=Rand()%vmax+1,y=Rand()%vmax+1;
				printf("%d
",Sum(l,r,x,y));
				break;
			}
		}
	}

	return 0;
}
原文地址:https://www.cnblogs.com/SovietPower/p/13739898.html