FJUT 2019暑假第三次周赛 C - 郭先生的魔法阵

牛客原题，出题人zb背锅。

题意：关于魔法阵问题，选择最小的代价并且满足题目所需的要求；

思路: 从小到大枚举每个能量大小，计算出破坏魔法阵需要的代价，取最小的就好了，因为所需要的魔法币最多只有200，所以开两个大小为200的数组暴力枚举代价就好了。详细过程看代码注释。

#include <ctime>
#include <map>
#include <set>
#include <list>
#include <cmath>
#include <stack>
#include <queue>
#include <vector>
#include <string>
#include <cstdio>
#include <bitset>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define lson rt << 1,l,m
#define rson rt << 1 | 1 ,m + 1, r
#define accept return 0
#define  mes(a, b)  memset(a, b, sizeof a)
typedef unsigned long long int ull;
typedef long long int ll;
const int    maxn = 1e6 + 10;
const int    maxm = 1e5 + 10;
// const int    mod  = 1e9 + 7;
const ll     INF  = 1e18 + 100;
const int    inf  = 0x3f3f3f3f;
const double pi   = acos(-1.0);
const double eps  = 1e-8;
using namespace std;
ll need[300],nneed[300];
ll cost = 0;
ll minn,n;

struct node{
    ll h,w,n;
    bool operator <(const node & a)const{
        return h < a.h;//按能量大小排序
    }
}t[maxn];

void solve(){
    mes(need,0);
    mes(nneed,0);
    sort(t + 1 , t + 1 + n);
    ll nown = 0;
    minn = INF;
    ll ans = 0;
    ll nowin = 0;
    for(int i = 1;i <= n;i++){   //从小到大枚举能量
        ans = 0;
        nneed[t[i].w] += t[i].n; //nneed数组存储当前拥有的魔法棒的所有能量的数量
        nown += t[i].n;    //当前拥有的所有的魔法棒的数量
        nowin += t[i].n;    //当前能量的魔法棒的数量
        cost -= t[i].n * t[i].w;  //破坏比当前魔法棒能量大的魔法棒所需的总花费
       if(t[i].h != t[i + 1].h){
            ans += cost;
            if(nown - (nowin * 2 - 1) > 0){
                ll left = nown - (nowin * 2 - 1);  //当前能量魔法棒数量严格大于当前拥有的所有魔法棒
                for(int j = 1;j <= 200;j++){
                    if(left > need[j]) {ans += j * need[j]; left -= need[j];} 
                    //减去比当前魔法棒能量小的魔法棒的最小花费的魔法棒
                    else{ans += j * left;break;}
                }
            }
            minn = min(minn,ans);
            for(int j = 1;j <= 200;j++) need[j] = nneed[j];
            nowin = 0;
        }
    }
    printf("%lld\n",minn);
}

int main(){
    while(cin >> n){
        cost = 0;
        for(int i = 1;i <= n;i++) {
            cin >> t[i].h >> t[i].w >> t[i].n;
            cost += t[i].w * t[i].n; //预处理最大的能量花费
        }
        solve();
    }
}