Acwing2024蓝桥杯多路归并

题目:

AcWing 1262. 鱼塘钓鱼

#include<iostream>
#include<cstring>
using namespace std;
const int N=105;
int n,a[N],d[N],timee[N],T,cost[N];
//get函数用于获取当前第x个鱼塘的最大数
int get(int x){
    return max(0,a[x]-d[x]*cost[x]);
}
//work函数(终点,剩余可以钓鱼的时间)
int work(int last,int rest){
    int temp=0;
    //清空cost数组
    memset(cost,0,sizeof(cost));
    //选rest个
    for(int k=1;k<=rest;k++){
        //每次找last个鱼塘中最大的数
        int i=1;
        for(int j=2;j<=last;j++){
            if(get(i)<get(j)){
                i=j;
            }
        }
        //加上所选钓鱼数,并且该鱼塘花费时间加一
        temp+=get(i);
        cost[i]++;
    }
    return temp;
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++) cin>>d[i];
    for(int i=2;i<=n;i++) {cin>>timee[i];timee[i]+=timee[i-1];} //前缀和
    cin>>T;
    int ans=0;
    //枚举所有终点的可能情况,并取最大值
    for(int i=1;i<=n;i++) ans=max(ans,work(i,T-timee[i]));
    cout<<ans<<endl;
    return 0;
}

AcWing 4656. 技能升级(第十三届省赛)

暴力归并 未AC(40 points)

#include<iostream>
using namespace std;
const int N=1e5+5;
int n,m,a[N],d[N],flag[N];
long long get(int x){
    return max(0,a[x]-d[x]*flag[x]);
}
long long work(int last){
    long long temp=0;
    for(int k=1;k<=m;k++){
        int i=1;
        for(int j=2;j<=last;j++){
            if(get(i)<get(j)){
                i=j;
            }
        }
        temp+=get(i);
        flag[i]++; 
    }
    return temp;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i]>>d[i];
    long long ans=work(n);
    cout<<ans<<endl;
    return 0;
}

优化:二分思想+多路归并 AC

#include<iostream>
using namespace std;
const int N=1e5+5;
int n,m,a[N],d[N];
bool check(int x){
    long long temp=0;
    for(int i=1;i<=n;i++){
        if(a[i]>=x){
            temp+=(a[i]-x)/d[i]+1;
        }
    }
    return temp>=m;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i]>>d[i];
    //二分出排第m大的数r
    int l=0,r=1e6;
    while(l<r){
        int mid=(l+r+1)/2;
        if(check(mid)) l=mid;
        else r=mid-1;
    }
    //ans记录所有>=r的数目之和,cnt记录>=r的数目
    long long ans=0,cnt=0;
    for(int i=1;i<=n;i++){
        if(a[i]>=r){
            int t=(a[i]-r)/d[i]+1;
            cnt+=t;
            int end=a[i]-(t-1)*d[i];
            ans+=(long long)(a[i]+end)*t/2;
        }
    }
    //题目求的是前m个数之和,因此要减去cnt-m个r
    cout<<ans-(cnt-m)*r<<endl;
    return 0;
}

vector操作+优先队列

vector是向量类型,可以容纳许多类型的数据,因此也被称为容器
头文件:#include<vector>
定义具有5个整型元素的向量,不具有初值,其值不确定
vector<int>a(5);
定义具有5个整型元素的向量，且给出的每个元素初值为1
vector<int>a(5,1);
常用函数:
a[i];                                 //返回a的第i元素,当且仅当a存在
a.back();                             //返回a的最后一个元素
a.front();                            //返回a的第一个元素
a.clear();                            //清空a中的元素=
a.empty();                            //判断a是否为空，空则返回true，非空则返回false
a.pop_back();                         //删除a向量的最后一个元素
a.push_back(5);                       //在a的最后一个向量后插入一个值为5的元素
a.size();                             //返回a中元素的个数
int b[3]={1,2,3};vector<int>a(b,b+3); //从数组中获得初值


升序队列
priority_queue<int,vector<int>,greater<int>>pq;
降序队列
priority_queue<int,vector<int>,less<int>>pq;
头文件:#include<queue>
常用函数
pq.top();   //访问队头元素
pq.empty(); //队列是否为空
pq.size();  //返回队列内元素个数
pq.push();  //插入元素到队尾(并排序)
pq.pop();   //弹出队头元素

P1090 [NOIP2004 提高组] 合并果子 / [USACO06NOV] Fence Repair G

#include<iostream>
#include<queue>
using namespace std;
priority_queue<int,vector<int>,greater<int>>pq;
int n,ans;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        pq.push(x);
    }
    while(pq.size()>1){
        int t1=pq.top();pq.pop();
        int t2=pq.top();pq.pop();
        ans+=t1+t2;
        pq.push(t1+t2);
    }
    cout<<ans<<endl;
    return 0;
}

Acwing 62. 丑数

class Solution {
public:
    int getUglyNumber(int n) {
        if(n==1) return 1;
        vector<int>a(1,1);
        int flag[1005],b[3]={2,3,5};
        memset(flag,0,sizeof(flag));
        while(a.size()<n){
            int temp=2e9;
            for(int i=0;i<3;i++){
                temp=min(temp,a[flag[i]]*b[i]);
            }
            a.push_back(temp);
            for(int i=0;i<3;i++){
                if(temp==a[flag[i]]*b[i]){
                    flag[i]++;
                }
            }
        }
        return a.back();
    }
};

AcWing 1378. 谦虚数字

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int k,n,a[105],flag[105];
int main(){
    cin>>k>>n;
    for(int i=0;i<k;i++) cin>>a[i];
    sort(a,a+k);
    vector<int>ans(1,1);
    while(ans.size()<=n){
        int temp=2e9;
        for(int i=0;i<k;i++){
            temp=min(temp,ans[flag[i]]*a[i]);
        }
        ans.push_back(temp);
        for(int i=0;i<k;i++){
            if(ans[flag[i]]*a[i]==temp){
                flag[i]++;
            }
        }
    }
    cout<<ans.back()<<endl;
    return 0;
}

AcWing 146. 序列

#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
const int N=2005;
int t,m,n,a[N],b[N],c[N];
typedef pair<int,int>pii;//<sum,id>当前和,当前下标
void merge(){
    priority_queue<pii,vector<pii>,greater<pii>>pq;
    //形成第一组数据
    for(int i=0;i<n;i++) pq.push({a[0]+b[i],0});
    //找n个数
    for(int i=0;i<n;i++){
        //存储最小值后删除
        pii temp=pq.top();
        pq.pop();
        int sum=temp.first,id=temp.second;
        //c数组临时存和
        c[i]=sum;
        //插入被删除的下一个数
        pq.push({sum-a[id]+a[id+1],id+1});
    }
    //利用c更新a,使a记录最终结果
    for(int i=0;i<n;i++) a[i]=c[i];
}
int main(){
    ios::sync_with_stdio(false);
    cin>>t;
    while(t--){
        cin>>m>>n;
        //输入第一行序列,并进行排序
        for(int i=0;i<n;i++) cin>>a[i];
        sort(a,a+n);
        //共合并m-1次
        for(int i=0;i<m-1;i++){
            //存储要合并的序列
            for(int j=0;j<n;j++) cin>>b[j];
            //合并a序列和b序列,并且把合并后的序列存到a中
            merge();
        }
        //输出a中前n项
        for(int i=0;i<n;i++) cout<<a[i]<<" ";
        cout<<endl;
    }
    return 0;
}