Wannafly挑战赛15 A-最小化价格

链接：https://www.nowcoder.com/acm/contest/112/A
来源：牛客网

现有n组人，m个地点，给出每组人的人数，每个地点可容纳的最大人数和选择的价格

要求一种方式，使得每组人都到一个各不相同的地点，最小化选择的价格

每个队伍的人都要在同一个地方每个地方只能有一个队伍

输入：

3 4
2 3 4
1 2
2 3
3 4

4 5

输出：

12

被自己菜醒：

线段树乱搞。

 

/**
把b按价值从小到大排序，如果能装队伍就装队伍，
用树状数组记录，二分查找是否有值。
*/
struct Node{
    int w,v,s;
}b[maxn+5];
int n,m,a[maxn+5],w[maxn+5],C[maxn+5];
bool cmp(Node n1,Node n2)
{
    if(n1.v == n2.v) return n1.w < n2.w;
    return n1.v < n2.v;
}
int lowbit(int x){return x&(-x);}
int Sum(int x)
{
    int sum = 0;
    while(x>0){
        sum+=C[x];
        x-=lowbit(x);
    }
    return sum;
}
void Add(int x,int d)
{
    while(x<=maxn) C[x]+=d,x+=lowbit(x);
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]), w[a[i]]++, Add(a[i],1);
    for(int i=1; i<=m; i++) scanf("%d %d",&b[i].w,&b[i].v);
    sort(b+1,b+m+1,cmp);
    ll ans = 0, cnt=0;
    for(int i=1;i<=m;i++)
    {
        int num = Sum(b[i].w);
        if(num==0) continue;
        int L = 0,R = b[i].w,mid;
        while(R-L>1)
        {
            mid = (R+L)>>1;
            if(Sum(mid)==num) R = mid;
            else L = mid;
        }
        ans+=b[i].v;
        cnt++;
        Add(R,-1);
    }
    if(cnt<n) ans = -1;
    printf("%lld\n",ans);
    return 0;
}

题解：（已经被自己菜醒！）

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
typedef long long int ll;
const int inf = 1e9;
const int maxn = 100000;
struct Node
{
    int w,v;
    bool operator < (const Node &N) const
    {
        return w > N.w;
    }
} b[maxn+5];
int n,m,a[maxn+5],mi[maxn*4];
bool cmp(int x,int y)
{
    return x>y;
}
priority_queue<int>q;
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]);
    for(int i=1; i<=m; i++) scanf("%d %d",&b[i].w,&b[i].v);
    sort(a+1,a+n+1,cmp);//a从大到小排序保证正确性
    sort(b+1,b+m+1);
    ll ans = 0;
    for(int i=1,j=1; i<=n; i++)
    {
        while(j<=m&&b[j].w>=a[i]) q.push(-b[j].v), j++;//这样队列里都是人数大于a[i]的。
        if(q.empty())
        {
            ans = -1;
            break;
        }
        ans-=q.top();
        q.pop();
    }
    printf("%lld\n",ans);
    return 0;
}