codeforces 169 div2 C

http://codeforces.com/problemset/problem/276/C

比赛的时候用的线段树。裸的吧。还wa了3次。。

//lazy_tag
//区间求和操作
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define MAXN 200010
int add[MAXN<<2];
long long  a[MAXN];
long long ans[MAXN];
void PushDown(int rt,int m)//m是父亲区间长度
{
    if(add[rt])
    {
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];
        add[rt]=0;
    }
}

//update
void update(int L,int R,int c,int l,int r,int rt)
{
    if (L<= l && r<=R)    //节点区间[l,r]被目标区间[L,R]包含
    {
        add[rt]+=c;
        if(L==R && c==0)
        {
            ans[L]+=add[rt];
            add[rt]=0;
        }
        return ;
    }
    PushDown(rt,r-l+1);
    int m = (l+r)>>1;
    if(L<=m)
        update(L,R,c,lson);
    if(m<R)
        update(L,R,c,rson);
}

int main()
{
    int N,Q;
    while(scanf("%d%d",&N,&Q)!=EOF)
    {
        memset(ans,0,sizeof(ans));
        memset(add,0,sizeof(add));
        for(int i=0;i<N;i++)
            scanf("%d",&a[i]);
        sort(a,a+N);
        while(Q--)
        {
            int l,r;
            scanf("%d%d",&l,&r);
            update(l,r,1,1,N,1);
        }
        for(int i=1;i<=N;i++)
            update(i,i,0,1,N,1);
        sort(ans+1,ans+N+1);
        long long sum=0;
        for(int i=0;i<N;i++)
            sum+=ans[i+1]*a[i];
        printf("%I64d\n",sum);
    }
    return 0;
}

不过看了别人的程序，觉得下面这种解法值得学习。ac了。以前好像看见过类似的问题，总是不知道怎么解决。作个保留。

// File Name: cc.cpp
// Author: Missa
// Created Time: 2013/2/25 星期一 23:13:05

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<string>
#include<vector>
#include<cstdlib>
#include<map>
#include<set>
using namespace std;
#define CL(x,v) memset(x,v,sizeof(x));
#define R(i,st,en) for(int i=st;i<en;i++)
#define ll long long
const int maxn = 2e5+5;
ll a[maxn],ans[maxn],w[maxn];
int n,m;

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        CL(a,0);CL(ans,0);CL(w,0);
        R(i,1,n+1)
            scanf("%I64d",&a[i]);
        R(i,0,m)
        {
            int l,r;
            scanf("%d%d",&l,&r);
            w[l]++;
            w[r+1]--;
        }
        int tmp=0;
        R(i,1,n+1)
        {
            tmp+=w[i];
            ans[i]=tmp;
        }
        sort(ans+1,ans+n+1);
        sort(a+1,a+n+1);
        ll sum=0;
        R(i,1,n+1)
            sum+=a[i]*ans[i];
        printf("%I64d\n",sum);
    }
    return 0;
}