免费送气球

免费送气球

http://acm.hdu.edu.cn/showproblem.php?pid=6464

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 537    Accepted Submission(s): 117

Problem Description

又到了GDUT一年一度的程序设计竞赛校赛的时间啦。同学们只要参加校赛，并且每解出一道题目就可以免费获得由ACM协会和集训队送出的气球一个。听到这个消息，JMC也想参加免费拿气球。可是，由于JMC太菜了而被禁止参赛，于是他找到你想让你帮忙参加比赛，可以通过执行下面的C++程序解决问题后获得气球并送给他。JMC保证了下面的程序一定能获得正确的结果。

void solve(int Q, int type[], long long first[], long long second[]) {
    vector<long long> vec;
    for (int i = 0; i < Q; ++i) {
        if (type[i] == 1) {
            long long k = first[i], val = second[i];
            while (k--) {
                vec.push_back(val);
            }
        }
        else if (type[i] == 2) {
            sort(vec.begin(), vec.end());
            long long l = first[i] - 1, r = second[i], res = 0;
            while (l < r) {
                res = (res + vec[l++]) % 1000000007;
            }
            printf("%lld ", res);
        }
    }
}


为防止你被JMC的代码搞到头晕目眩，JMC特意给出了问题的文字描述。已知一开始有一个空序列，接下来有Q次操作，每次操作给出type、first和second三个值。当type为1时，意味着该操作属于第一种操作：往序列尾部添加first个second数。当type为2时，意味着该操作属于第二种操作：查询序列中第first小至第second小的数值之和（一共有(second - first + 1)个数被累加），并将结果对1000000007取模后输出。

 

Input

单组数据
第一行一个Q（1 <= Q <= 1e5），代表Q次操作。
接下来有Q行，每行包含三个整数type、first和second；其中1 <= type <= 2。当type等于1时，0 <= first,second < 1e9。当type等于2时，1 <= first <= second，且first和second均不大于目前已添加进序列的数的数量。

 

Output

对于每次操作二，将结果对1000000007取模后输出。

 

Sample Input

6

1 5 1

1 6 3

2 2 5

2 4 8

1 2 2

2 4 8

 

Sample Output

4

11

9

 

思路：用权值线段树找第n小的数，用线段树求和

#include<bits/stdc++.h>
using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define sqr(x) ((x)*(x))
#define pb push_back
#define eb emplace_back
#define maxn 100005
#define eps 1e-8
#define pi acos(-1.0)
#define rep(k,i,j) for(int k=i;k<j;k++)
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<double,double>pdd;
typedef pair<int,char> pic;
typedef pair<pair<int,string>,pii> ppp;
typedef unsigned long long ull;
const long long MOD=1000000007;
const double oula=0.57721566490153286060651209;
using namespace std;

struct sair{
    int type;
    ll x,y;
}a[maxn];

vector<ll>ve;

ll num[maxn<<2],sum[maxn<<2];

void push_up(int rt){
    num[rt]=num[rt<<1]+num[rt<<1|1];
    sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%MOD;
}

void update(int L,ll nu,ll v,int l,int r,int rt){
    if(l==r){
        num[rt]+=nu;
        sum[rt]=(sum[rt]+(v*nu)%MOD)%MOD;
        return;
    }
    int mid=l+r>>1;
    if(L<=mid) update(L,nu,v,lson);
    else update(L,nu,v,rson);
    push_up(rt);
}

ll querynum(ll L,int l,int r,int rt){///L忘了开LL,找了半天bug。。。。
    if(l==r) return r;
    int mid=l+r>>1;
    if(num[rt<<1]>=L) return querynum(L,lson);
    else return querynum(L-num[rt<<1],rson);
}

ll querynum(int L,int R,int l,int r,int rt){
    if(L<=l&&R>=r) return num[rt];
    int mid=l+r>>1;
    ll ans=0;
    if(L<=mid) ans+=querynum(L,R,lson);
    if(R>mid) ans+=querynum(L,R,rson);
    return ans;
}

ll querysum(int L,int R,int l,int r,int rt){
    if(L<=l&&R>=r) return sum[rt];
    int mid=l+r>>1;
    ll ans=0;
    if(L<=mid) ans=(ans+querysum(L,R,lson))%MOD;
    if(R>mid) ans=(ans+querysum(L,R,rson))%MOD;
    return ans;
}


int getid(ll x){
    return lower_bound(ve.begin(),ve.end(),x)-ve.begin()+1;
}

int main(){
    std::ios::sync_with_stdio(false);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i].type>>a[i].x>>a[i].y;
        if(a[i].type==1) ve.pb(a[i].y);
    }
    sort(ve.begin(),ve.end());
    ll pos,pos1,pos2,tmp1,tmp2;
    for(int i=1;i<=n;i++){
        if(a[i].type==1){
            pos=getid(a[i].y);
            update(pos,a[i].x,a[i].y,1,n,1);
        }
        else{
            ll ans=0;
            pos1=querynum(a[i].x,1,n,1);///ve[pos1-1]为值
            tmp1=querynum(1,pos1,1,n,1);
            pos2=querynum(a[i].y,1,n,1);
            if(pos2>1) tmp2=querynum(1,pos2-1,1,n,1);
            else tmp2=0;
           // cout<<pos2<<endl;
          //  cout<<pos2-1<<endl;
           // tmp2=querynum(1,pos2-1,1,n,1);
            /**
                1.x==y
                2.x与y在同一个区域
                3.x与y相邻
            */
            if(pos1==pos2){
                ans=((a[i].y-a[i].x+1)*ve[pos1-1])%MOD;
            }
            else if(pos2-pos1==1){
                ans=(((tmp1-a[i].x+1)*ve[pos1-1])%MOD+((a[i].y-tmp2)*ve[pos2-1])%MOD)%MOD;
            }
            else{
                ans=(((tmp1-a[i].x+1)*ve[pos1-1])%MOD+((a[i].y-tmp2)*ve[pos2-1])%MOD+querysum(pos1+1,pos2-1,1,n,1)%MOD)%MOD;
            }

            cout<<ans<<endl;
        }
    }
}
/**
6
1 5 1000000000
1 6 3000000000
2 2 5
2 4 8
1 2 2000000000
2 4 8
*/