C.哦~唔西迪西小姐~(贪心,stl)

目录

题目描述

输入描述:

输出描述:

输入

输出

输入

输出

思路:

代码:

---

题目描述

唔西迪西现在正处在一个冰火迷宫中，迷宫由 n 个格子组成，每个格子要么是冰之格，要么是火之格，唔西迪西刚开始可以选择从迷宫中任意一个开始走，走到第 i 个位置时会得到值为 ai的积分。（注意：唔西迪西也可以选择一个格子都不走）

如果唔西迪西当前在冰之格，那么她可以选择一个编号大于当前格子的冰之格，跳到那里。如果唔西迪西当前在火之格，那么她可以选择一个编号大于当前格子的火之格，跳到那里。如果唔西迪西目前没有格子可以走，那么结束。同时，即使存在可以跳过去的格子，唔西迪西也可以选择在任意时刻结束。

唔西迪西想最大化她的得分，于是她学会了一个超能力，她能在比赛开始的时候改变最多 mmm 个格子的状态，即将一个格子从冰之格变成火之格或从火之格变成冰之格，改变第 iii 个格子的状态会让唔西迪西的得分减少 pip_ipi​ 。（唔西迪西改变格子的状态后才开始挑选起点开始行动，也就是说，得分分成两部分，一部分是改变格子状态的得分，一部分是走格子的得分）

你能告诉唔西迪西她最多得几分吗

输入描述:

第一行两个正整数 n,m (1≤n,m≤105)。

第二行 n个整数表示 ai​。

第三行 n个整数表示 pi (−105≤ai​,pi​≤105)。

第四行 n个整数 bi为 0 或 1，0 表示这个格子是冰之格，1 表示这个格子是火之格。

输出描述:

一个数表示答案。

示例1

输入

3 2
1 15 9
2 5 7
1 0 1

输出

20

示例2

输入

10 3
80 86 57 69 59 52 94 74 9 63
8 32 77 64 53 49 22 68 27 63
1 0 0 0 1 0 0 1 1 1

输出

442

思路:

1,每个位置只能用一次;p[i]可能为负,所以即便a[i]>0,也有可能不是最优解

2,冰(火同理),分类,首先,本身是冰,本身大于0,当然也可加,也可翻转,本身不大于零,翻转

其次,本身不是冰,选max(-p[i],a[i]-p[i]);去翻转

代码:

#include<bits/stdc++.h>
using namespace std;
#define int long long//开对数据类型
const int maxj=1e5+100;
typedef long long ll;//冰火,多种策略,判断逻辑关系.
#define pu push_back
int a[maxj],p[maxj],b[maxj];//p[i]可为负
bool cmp(int a,int b){
    return a>b;
}
int32_t main(){
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;++i)
    cin>>a[i];
    for(int i=1;i<=n;++i){
    cin>>p[i],p[i]=-p[i];//变负为正
    }
    for(int i=1;i<=n;++i)
        cin>>b[i];
    int ans=0;
    for(int i=0;i<=1;++i){
        int res=0;
        vector<int>v;
        for(int j=1;j<=n;++j){
            if(b[j]==i){
            if(a[j]>0){
            res+=a[j];
            if(-a[j]+p[j]>0)
                v.pu(-a[j]+p[j]);
        }
            else{
                if(p[j]>0)v.pu(p[j]);
            }
        }
        else{
            v.pu(max(a[j]+p[j],p[j]));
        }
        }
        sort(v.begin(),v.end(),cmp);//排序选最大
        int d;
        if(m<v.size())d=m;
        else d=v.size();
        for(int j=0;j<d;++j){
            if(v[j]>0)res+=v[j];
        }
        ans=max(ans,res);
    }
    cout<<ans<<'\n';
    return 0;
}