C. Arthur and Table

链接

[https://codeforces.com/contest/557/problem/C]

题意

给你每个木棍的长度，以及移除每个木棍的代价
使得桌子稳定，最长的木棍必须大于剩下的一半

分析

这个数据范围，就是得暴力和前缀和还有一些技巧
枚举每一种长度，大于它的全部移除，
那么剩下的就尽可能地移除少量的木棍，而且移除可以移除的最小代价
复杂度2e7左右。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
struct str{
    int l,d;
}a[maxn];
int num[maxn];//长度为i的数量 
int prenum[maxn];//长度小于等于i的有多少个 
int s[maxn];// 第i个之前全部移除需要的代价 
int t[210];//当前代价为i的有多少个 
bool cmp(str x,str y){
    if(x.l==y.l) return x.d<y.d;
    return x.l<y.l;
}

int main(){
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    int n;
    while(cin>>n){
        memset(num,0,sizeof(num));
        memset(prenum,0,sizeof(prenum));
        memset(s,0,sizeof(s));
        memset(t,0,sizeof(t));
        for(int i=1;i<=n;i++)
        {
            cin>>a[i].l;
            num[a[i].l]++;
        }
        
        for(int i=1;i<=n;i++)
        cin>>a[i].d;
        
        sort(a+1,a+n+1,cmp);
         
         for(int i=1;i<=n;i++)
         s[i]=s[i-1]+a[i].d;
         
         for(int i=1;i<=100000;i++)
         prenum[i]=prenum[i-1]+num[i];
         int ans=1e9;
         
         for(int i=1;i<=100000;i++)
         {
            if(!num[i]) continue;
            int tem=prenum[i]-(2*num[i]-1);//该移除的个数 
            int cur=s[n]-s[prenum[i]];
            //cout<<i<<' '<<tem<<' '<<cur<<endl;
                for(int j=1;j<=200&&tem>0;j++)
                {
                    if(tem>=t[j]){
                        tem-=t[j];
                        cur+=t[j]*j;
                     }
                     else {
                        cur+=tem*j;
                        tem=0;
                     }
                 }
             ans=min(ans,cur);
             for(int j=prenum[i-1]+1;j<=prenum[i];j++)
             t[a[j].d]++;
            
         }
         cout<<ans<<endl;
    }
    return 0;
}