这道题比赛时候没做出来,下来一看才发现是排序傻逼题。
把每个偏好的人做成一个vector,从大到小排序,做一个前缀和。然后将每种人数做一个桶,在桶里装每种科目选择人数为i的时候分数总和。
遍历每一维vector,把各个位置上面的vector加到sum数组中,最后sum数组里面挑出最大值。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<int> vec[100010];//vector一定要声明类型
ll sum[100010];
bool cmp(int a,int b)
{
return a>b;
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
int subject,score;
scanf("%d%d",&subject,&score);
vec[subject].push_back(score);
}
for(int i=1;i<=m;i++)
{
sort(vec[i].begin(),vec[i].end(),cmp);
}
int maxsize=0;
for(int i=1;i<=m;i++)
{
for(int j=1;j<vec[i].size();j++)
{
vec[i][j]+=vec[i][j-1];
}
maxsize=max((int)vec[i].size(),maxsize);
}
ll res=0;
for(int i=1;i<=m;i++)
{
for(int j=0;j<vec[i].size();j++)
{
if(vec[i][j]>0)
sum[j+1]+=vec[i][j];
}
}
for(int i=1;i<=maxsize;i++)
{
//printf("%d ",sum[i]);
res=max(sum[i],res);
}
printf("%lld\n",res);
}