[Educational Codeforces Round 55 (Rated for Div. 2)][C. Multi-Subject Competition]

https://codeforc.es/contest/1082/problem/C

题目大意:有m个类型,n个人,每个人有一个所属类型k和一个能力v,要求所选的类型的人个数相等并且使v总和最大(n,m<=1e5)

题解:用vector存下每种类型的各个v并且每种类型根据v按从大到小排序,然后处理出每种类型的前缀和,然后扫每种类型的所有前缀和,如果该类型在i处的前缀和大于0,则相应的ans[i]加上这个类型在i处的前缀和,最后求出max(ans[i])(1<=i<=n)即可.

注意:这题的外层循环应该是遍历m个类型,然后扫每种类型的前缀和来更新ans[i],这样子因为一共n个人,所以最多n次扫描过程复杂度为O(n),这种方式由于是根据每个类型的size来遍历的会减少很多无用的遍历,而如果外层循环是长度n,然后根据长度来更新ans[i],这样子会无视类型长度,进行很多次无用的遍历,复杂度是O(m*n),显然会T掉

#include <iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<stack>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
const int mod = 1e9+7;
vector<int>vc[100005];
vector<ll>sum[100005];
ll ans[100005];
int main()
{
  // freopen("in.txt","r",stdin);
    int n,m;
    cin>>n>>m;
    int aa,bb;
    int maxx=1;
    for(int i=0;i<n;i++){
        scanf("%d%d",&aa,&bb);
        vc[aa].push_back(bb);
        maxx=max(maxx,aa);
    }
    int minn=0;//
    for(int i=1;i<=maxx;i++){
        minn=max(minn,(int)vc[i].size());
        sort( vc[i].begin(),vc[i].end(),greater<int>() );
        for(int j=0;j<vc[i].size();j++){

            if(j)sum[i].push_back(sum[i][j-1]+vc[i][j]);
            else {sum[i].push_back(vc[i][j]);}
        }
    }
    /*for(int i=1;i<=minn;i++){
        for(int j=1;j<=maxx;j++){
            if(vc[j].size()>=i&&sum[j][i-1]>=0){
                ans[i]+=sum[j][i-1];
            }
        }
    }*///错误的扫描方式,会T
    for(int i=1;i<=maxx;i++){
        for(int j=0;j<vc[i].size();j++){
            if(sum[i][j]>=0){
                ans[j+1]+=sum[i][j];
            }
        }
    }//正确的扫描方式
    ll aans=0;
    for(int i=1;i<=minn;i++){
        aans=max(ans[i],aans);
    }
    cout << aans<<endl;
    return 0;
}