正题
题目链接:https://www.luogu.com.cn/problem/P3168
题目大意
nnn个任务(si,ei,ti)(s_i,e_i,t_i)(si,ei,ti)表示从si∼eis_i\sim e_isi∼ei的任务优先级为tit_iti。
然后每次询问一个时刻前kkk小的优先级任务优先级和。
解题思路
开主席树,在sis_isi处让tit_iti加一,ei+1e_i+1ei+1处让tit_iti减一。
然后直接查询,需要注意的是查询到最后一个时不能直接返回sumxsum_xsumx,因为需要返回的是前kkk的和。
时间复杂度O(nlogn)O(n\log n)O(nlogn)
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define ll long long
using namespace std;
const ll N=1e5+10,inf=1e7;
ll n,m,cnt,rt[N];
ll w[N<<7],ls[N<<7],rs[N<<7],sum[N<<7];
vector<int> q[N];
ll Change(ll &x,ll l,ll r,ll pos,ll val){ll now=++cnt;w[now]=w[x]+val;sum[now]=sum[x]+val*pos;if(l==r)return now; ll mid=(l+r)>>1;if(pos<=mid)ls[now]=Change(ls[x],l,mid,pos,val),rs[now]=rs[x];else ls[now]=ls[x],rs[now]=Change(rs[x],mid+1,r,pos,val);return now;
}
ll Ask(ll x,ll l,ll r,ll k){if(w[x]<=0)return 0;if(l==r)return k*l;ll mid=(l+r)>>1;if(k<=w[ls[x]])return Ask(ls[x],l,mid,k);return Ask(rs[x],mid+1,r,k-w[ls[x]])+sum[ls[x]];
}
int main()
{
// printf("%d",sizeof(sum)/1024/1024);scanf("%lld%lld",&n,&m);for(ll i=1;i<=n;i++){ll l,r,w; scanf("%lld%lld%lld",&l,&r,&w);q[l].push_back(w); q[r+1].push_back(-w);}for(ll i=1;i<=1e5;i++){if(q[i].empty()) rt[i]=rt[i-1];else rt[i]=Change(rt[i-1],1,inf,abs(q[i][0]),(q[i][0]>0)?1:-1);for(ll j=1;j<q[i].size();j++)rt[i]=Change(rt[i],1,inf,abs(q[i][j]),(q[i][j]>0)?1:-1);}ll last=1;for(ll i=1;i<=m;i++){ll w,x,y,z;scanf("%lld%lld%lld%lld",&w,&x,&y,&z);x=1+(x*last+y)%z;printf("%lld\n",last=Ask(rt[w],1,inf,x));}return 0;
}