CodeForces - 1019A Elections(暴力枚举+思维)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_37129433/article/details/82958206

Elections
题 意:有n个选民，m个政党，每个选民都有自己要投的政党，和可以被贿赂的金额， $pi,ci$。现在1号政党想赢的这场选举，问最少需要花费多少金钱。
数据范围:
$1<=n,m<=3000$
$1<=pi<=m$
$1<=ci<=1e9$
输入样例:

1 2
1 100

输出样例:

0

输入样例:

5 5
2 100
3 200
4 300
5 400
5 900

输出样例:

500

思 路:n,m只有3000，所以可以去枚举最后稳赢的选票。其实也就是这一个地方难想一点。其他地方都很好写的。一点点思维。
收 获:增强了枚举的意识了吧。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<cmath>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<11
#define IN freopen("input.txt","r",stdin)
#define debug(x) cout<< #x <<" = "<< (x) <<endl;
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const int maxn = 3e3+5;
int vis[maxn];
int n,m;
struct node {
    ll p,c,id;
    bool operator<(const node &_p)const {
        return c<_p.c;
    }
} a[maxn];
vector<node> vec,vec1;
int mp[maxn];
int mp1[maxn];
ll solve(int x) {
    memset(vis,0,sizeof(vis));
    for(int i=1; i<=m; i++)mp1[i]=mp[i]
    vec.clear();
    vec1.clear();
    for(int i=1; i<=n; i++) {
        if(a[i].p == 1)continue;
        if(mp1[a[i].p] >= x) {
            vec.push_back(node{a[i].p,a[i].c,i});
        } else {
            vis[i] = 1;
            vec1.push_back(node{a[i].p,a[i].c,i});
        }
    }
    sort(vec.begin(),vec.end());
    ll sum = 0;
    int res = mp1[1];
    for(int i=0; i<vec.size(); i++) {
        ll temp1 = vec[i].p,temp2 = vec[i].c,id = vec[i].id;
        if(mp1[temp1] < x) {
            if(!vis[id]) {
                vec1.push_back(node{temp1,temp2,id});
            }
            continue;
        }
        sum += temp2;
        mp1[temp1]--;
        res++;
    }
    sort(vec1.begin(),vec1.end());
    for(int i=0; i<vec1.size(); i++) {
        if(res<x) {
            res++;
            sum+=vec1[i].c;
        } else {
            break;
        }
    }
    if(res!=x) return INF;
    else return sum;
}
int main() {

#ifdef LOCAL
    IN;
#endif
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++) {
        scanf("%lld %lld",&a[i].p,&a[i].c);
        mp[a[i].p]++;
    }
    ll ans = INF;
    for(int i=mp[1]; i<=n; i++) {
        ans = min(solve(i),ans);
    }
    printf("%lld\n",ans);
    return 0;
}