题目链接:https://cn.vjudge.net/problem/CodeForces-1082C
题目大意:
给定n名选手,每名选手都有唯一选择的科目si和对应的能力水平。并且给定科目数量为m。求选定若干个科目,并且每个科目参与选手数量相同的情况下的最大能力水平。
解题思路:
首先将每个选手的能力值插入到对应的科目里面,用优先队列维护,求前缀和,不断加入的答案数组里(为负值时不加入),最后遍历答案数组求最大值即可
AC代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int num[100005];
priority_queue<int> q[100005];
vector<int> v[100005];
int ss[100005] = {0};
int main()
{
int n, m, i, j, x, y, sum, ans = 0;
scanf("%d%d",&n,&m);
for(i=1;i<=m;i++)
{
num[i] = 0;
}
for(i=1;i<=n;i++)
{
scanf("%d %d",&x,&y);
num[x]++;
q[x].push(y);
}
for(i=1;i<=m;i++)
{
sum = 0;
for(j=0;j<num[i];j++)
{
sum += q[i].top();
q[i].pop();
v[i].push_back(sum);
}
}
for(i = 1; i <= m; i++)
{
ll len = num[i];
if(len > 0)
{
for(j = 0; j < len; j++)
{
if(v[i][j] > 0)
{
ss[j + 1] += v[i][j];
}
}
}
}
for(i = 1; i <= n; i++)ans = max(ans, ss[i]);
printf("%d\n",ans);
return 0;
}