Educational Codeforces Round 91 (Rated for Div. 2) D. Berserk And Fireball

Educational Codeforces Round 91 (Rated for Div. 2) D. Berserk And Fireball_第1张图片
题目链接
题目大意:有一个长度为n的数组a和一个长度为m的数组b。你可以执行两种操作,花费x消除a中连续k个数,或者花费y消除a中连续两个数中较小的那个数。问最少花费多少能从a中得到b。不能则输出-1.
思路:因为最近训练的缘故,思路很快就有了。
首先只有三种情况:1.都用方案1,有剩余则对剩余的使用方案二。
2.都用方案2.
3.在2的基础上可能会出现区间内最大值大于两个端点。所以我们可以用方案一消除那些最大值,最终也能消除掉区间内的所有数字。当然这个前提是这个区间长度大于等于k。
求最大值直接 O ( n ) O(n) O(n)去求就行,时限2s,处理各个区间也是 O ( n ) O(n) O(n),因为中间那个while循环的缘故是不会超时的
思路有了就写代码。注意开long long。代码很好写。具体细节看代码把。(我个人写的可能比较麻烦,我是把最左边和最右边两个区间单独处理的)。

#include
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int N=2e5+10;
int n,m,x,y,k;
ll a[N],b[N],p[N];
void solve()
{
    int f,flag=0;
    ll ans=0,res;
    for(int i=1;i<m;i++)
    {
        res=inf;
        f=0;
        ll ma=0,len=0;
        for(int j=p[i]+1;j<=p[i+1]-1;j++) ma=max(ma,a[j]),len++;
        if(ma>max(a[p[i]],a[p[i+1]])&&len>=k) res=min(res,(len-k)*y+x),f=1;
        if(ma<max(a[p[i]],a[p[i+1]])) res=min(res,len*y),f=1;
        if(len>=k) res=min(res,len/k*x+len%k*y),f=1;
        if(!f)
        {
            flag=1;
            break;
        }
        ans+=res;

    }
    ll ma=0,len=0;
    f=0,res=inf;
    for(int j=1;j<=p[1]-1;j++) ma=max(ma,a[j]),len++;
    if(ma>a[p[1]]&&len>=k) res=min(res,(len-k)*y+x),f=1;
    if(ma<a[p[1]]) res=min(res,len*y),f=1;
    if(len>=k) res=min(res,len/k*x+len%k*y),f=1;
    if(!f)
    {
        cout<<-1<<endl;
        return;
    }
    ans+=res;
    f=0,res=inf;
    ma=0,len=0;
    for(int j=p[m]+1;j<=n;j++) ma=max(ma,a[j]),len++;
    if(ma>a[p[m]]&&len>=k) res=min(res,(len-k)*y+x),f=1;
    if(ma<a[p[m]]) res=min(res,len*y),f=1;
    if(len>=k) res=min(res,len/k*x+len%k*y),f=1;
    if(!f)
    {
        cout<<-1<<endl;
        return;
    }
    ans+=res;
    if(flag) cout<<-1<<endl;
    else cout<<ans<<endl;
}
int main()
{
    cin>>n>>m;
    cin>>x>>k>>y;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=m;i++) cin>>b[i];
    int pos=1,flag=0;
    for(int i=1;i<=m;i++)
    {
        while(a[pos]!=b[i]&&pos<=n) pos++;
        if(pos==n+1)
        {
            flag=1;
            break;
        }
        p[i]=pos;
        pos++;
    }
    if(flag) cout<<-1<<endl;
    else solve();
    return 0;
}

你可能感兴趣的:(codeforces,贪心算法)