GYM 100500 SCPC 14 C ICPC Giveaways 思维 前缀和

TP

题意: 给出n类共m个物品, 每类价值相同, 从中选出任意类, 这些类中每种取X个, 问在X取任意值的情况下, 总价值最多是多少?

思路:

按照每种物品的个数将物品排序, 前缀和处理出区间内每个物品取1个的总价值.
暴力从第一种物品扫到最后一种物品(for i: 1 to n), 区间宽度=前缀和数组sum[n]-sum[i-1], 高度是num[i].

换句话说, 从第i种物品取到第n种物品, 每种只取1个的价值由前缀和确定, 而每个种类到底能取多少个呢? 这就由出现个数最少的种类决定了. 而这个数组是按num排序的, 所以每个种类取num[i]即为最优. 最后让每个1 to n 的数都当作i, 取最大值就是最优解.

 

代码:

#include<bits/stdc++.h>
#define fuck(x) std::cout<<"["<<#x<<"->"<<x<<"]"<<endl;
using namespace std;
typedef long long ll;

const int M=1e4+5;
const int inf=1e9+5;
const int mod=1e9+7;

int m,n,c;


struct node {
    int v,num;
} pep[M];

bool cmp(node a,node b) {
    return a.num<b.num;
}

int sum[M];

int main() {
    int _;
    scanf("%d",&_);
    int caz=0;
    while(_--) {
        memset(pep,0,sizeof(pep));
        scanf("%d%d%d",&m,&n,&c);
        for(int i=0; i<m; i++) {
            int temp;
            scanf("%d",&temp);
            pep[temp].num++;
            pep[temp].v=temp*temp%c;
        }
        sort(pep+1,pep+n+1,cmp);
        sum[0]=0;
        for(int i=1; i<=n; i++) {
            sum[i]=sum[i-1]+pep[i].v;
        }
        int ans=0;
        for(int i=1; i<=n; i++) {
            ans=max(ans,pep[i].num*(sum[n]-sum[i-1]));
        }
        printf("Case %d: %d\n",++caz,ans);
    }
    return 0;
}