hdu 4866：Shooting（主席树）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4866

题目大意：

在二维坐标轴上给你一些连续的线段，然后让你求个横坐标x处正上方前k个线段的价值。

解题思路：

题目要求强制在线，而且因为横坐标的问题，可以直接用主席树对每个横坐标建树来做，不得不说主席树做这种题目确实是很方便。这题有个小坑点就是 某个高度会有重合的线段，主席树递归到最后肯定是在一个高度嘛，对那个高度你应该能获得多少价值特判一下就好。题目真心不难，而且感觉自己代码跟网上比起来挺简洁的，但是因为一个巨傻逼的地方 我整整wa了两天，难受= = ，

Ac代码：

#include<bits/stdc++.h>
#define lson rt<<1
#define rson rt<<1|1
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
const int INF=1e9+7;
int dx[4]={0,1,0,-1};
int dy[4]={1,0,-1,0};
int n,m,x,p,cnt,tot,root[maxn];
ll ans;
char sk;
struct Node     //储存线段+1 -1 
{
    int pos,val;
    int flag;
    bool operator<(const Node &p) const
    {
        return pos<p.pos;
    }
}g[maxn*2];
struct node     //主席树结点  分别储存 当前一共有的线段数量和价值和
{
    int l,r,sv;
    ll sum;
}t[maxn*40];
vector<int> v;  //离散化
int getid(int x){return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}
void update(int l,int r,int &x,int y,int flag,int val,int pos)  //主席树更新操作
{
    t[++tot]=t[y],t[tot].sv+=flag,t[tot].sum+=(val*flag),x=tot;
    if(l==r)
        return ;
    int mid=(l+r)/2;
    if(pos<=mid)
        update(l,mid,t[x].l,t[y].l,flag,val,pos);
    if(pos>mid)
        update(mid+1,r,t[x].r,t[y].r,flag,val,pos);
}
ll query(int l,int r,int x,int k)       //查询
{
    if(l==r)
    {
        if(t[x].sv!=0)
            return (ll)t[x].sum/t[x].sv*min(t[x].sv,k);
        return 0;
    }
    int mid=(l+r)/2;
    int tmp=t[t[x].l].sv;       //这个地方刚开始int sg=t[t[x].l].sum wa成了**，引以为戒
    if(tmp>=k)
        return query(l,mid,t[x].l,k);
    else
        return t[t[x].l].sum+query(mid+1,r,t[x].r,k-tmp);
}
int main()
{
    while(scanf("%d%d%d%d",&n,&m,&x,&p)!=EOF)
    {
        v.clear();
        memset(root,0,sizeof root);
        memset(g,0,sizeof g);
        int cnt=0;tot=0;
        for(int i=1;i<=n;i++)
        {
            int l,r,dis;
            scanf("%d%d%d",&l,&r,&dis);
            g[++cnt].pos=l,g[cnt].flag=1,g[cnt].val=dis;    //把线段的左右端点打上标记，注意r+1
            g[++cnt].pos=r+1,g[cnt].flag=-1,g[cnt].val=dis; 
            v.push_back(dis);
        }
        sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());
        sort(g+1,g+1+cnt);  //按所在位置排序
        int sk=1;
        for(int i=1;i<=x;i++)
        {
            int last=root[i-1];
            root[i]=root[i-1];
            while(g[sk].pos==i&&sk<=cnt)
            {
                update(1,n,root[i],last,g[sk].flag,g[sk].val,getid(g[sk].val)); //对每个端点对应更新
                last=root[i];
                sk++;
            }
        }
        ll pre=1;
        for(int i=1;i<=m;i++)
        {
            int ps,a,b,c;
            scanf("%d%d%d%d",&ps,&a,&b,&c);
            int k=((a%c*pre%c)+b)%c;
            ans=query(1,n,root[ps],k);  //查询答案即可
            if(pre>p)
                ans=ans*2;
            printf("%I64d\n",ans);
            pre=ans;
        }
    }
    //system("pause");
}