Code Forces 557 C. Arthur and Table(贪心)

来源:互联网 发布:大唐软件校园招聘 编辑:程序博客网 时间:2024/06/03 18:04

Description
给出一个桌子,有n个腿,每个腿的长度是l,拆掉这条腿的花费是d,当最长的腿占腿总数大于其他腿的总数,那么合法,问如何才能花费最小到达合法情况
Input
第一行为桌腿数量n,第二行n个整数表示每条桌腿的长度,第三行n个整数表示拆掉每条桌腿的花费
Output
输出达到合法情况的最小花费
Sample Input
6
2 2 1 1 3 3
4 3 5 5 2 1
Sample Output
8
Solution
枚举最后的最长的那条腿的方法,比这条腿长的腿一定是要拆掉的,所以可以预处理出比长度为i的腿拆掉的总数和总花费,然后在枚举每个i的时候就可以o(1)的获取到大于i的总数和总花费,然后在从比当前腿小的中选花费最小的,首先对所有的腿排序,然后按从小到大枚举的过程中,利用一个数组记录下每个d值下的个数(预处理),然后每次将比它小的添加,这个操作均摊O(n)的复杂度,然后每次需要枚举200值范围内从小到大,直到当前剩余的腿的个数小于最大的腿的个数,总的复杂度是201*O(n)
Code

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>using namespace std;#define maxn 111111struct node{    int l,d;}leg[maxn];int cnt[222];//记录每个d值的个数 int mark[maxn];//记录每个长度出现的次数 int tol[maxn];//记录拆掉所有大于等于某长度的桌腿的花费 int sum[maxn];//记录大于等于某长度的桌腿数量 int ll[maxn];//记录每条桌腿的长度 int dd[maxn];//记录拆掉每条桌腿的花费 int cmp(node a,node b){    return a.l<b.l;//将桌腿按长度排序 }int main(){    int n;    while(scanf("%d",&n)!=EOF)    {        memset(cnt,0,sizeof(cnt));        memset(mark,0,sizeof(mark));        memset(tol,0,sizeof(tol));        memset(sum,0,sizeof(sum));        int up=0,low=maxn;//up表示最长桌腿,low表示最短桌腿         for(int i=0;i<n;i++)        {            scanf("%d",&ll[i]);            up=max(up,ll[i]);            low=min(low,ll[i]);        }        for(int i=0;i<n;i++)            scanf("%d",&dd[i]);        for(int i=0;i<n;i++)        {            mark[ll[i]]++;//该长度桌腿数量加一             tol[ll[i]]+=dd[i];//拆掉所有大于等于该桌腿长度的花费         }        for(int i=up;i>=low;i--)        {            sum[i]=sum[i+1]+mark[i];//大于等于某长度的桌腿数量            tol[i]+=tol[i+1];//拆掉所有大于等于某长度的桌腿的花费         }        for(int i=0;i<n;i++)        {            leg[i].l=ll[i];            leg[i].d=dd[i];        }        sort(leg,leg+n,cmp);//对桌腿按长度升序排         int res=0;        int ans=1<<29;//花费         for(int i=low;i<=up;i++)        {            if(!mark[i])//没有该长度的桌腿                 continue;            while(res<n&&leg[res].l<i)//记录所有长度小于该桌腿的桌腿花费                 cnt[leg[res++].d]++;            int temp=tol[i+1];//temp初始化为拆掉所有大于当前桌腿长度桌腿的花费             int m=n-sum[i];//m为小于当前桌腿长点的桌腿数量             for(int j=1;j<=200;j++)//枚举所有长度             {                if(m<mark[i])//满足条件                     break;                if(m-mark[i]+1<cnt[j])//要切的桌腿数量小于该长度的桌腿数量                 {                    temp+=(m-mark[i]+1)*j;//要切的桌腿全部为该长度即可                     m-=mark[i]-1;                    break;                }                temp+=cnt[j]*j;//要切的桌腿数量大于等于该长度的桌腿数量                 m-=cnt[j];            }            ans=min(ans,temp);//更新最小花费         }        printf("%d\n",ans);    }    return 0;}
0 0
原创粉丝点击