码迷,mamicode.com
首页 > 其他好文 > 详细

CF1278C-Berry Jam-(前缀和)

时间:2019-12-21 09:17:45      阅读:74      评论:0      收藏:0      [点我收藏+]

标签:指示   越界   http   def   target   public   内容   多次   +=   

https://vjudge.net/problem/CodeForces-1278C

题意:有2n瓶果酱,中间有一个楼梯隔开,从中间往左或右两边清空果酱,使得两种果酱的数量相等,最少要清空多少瓶

思路:第一眼看到并没有什么思路,想用果酱数量的前缀和,但不知道怎么运用,看到一篇博客,把数字2改为-1这几个字眼,恍然大悟。


 例如第一组数据

原数据 1,1,1,2,2,         1,2,1,2,1,1,2

变为   1,1,1,-1,-1,1,   -1,1,-1,1,1,-1

前缀和 1,2,3,2,1,2,     0,1,0,1,0,-1

因为清除果酱只能从中间的楼梯向两边扩展,所以左边的前缀和是→累加的,右边的前缀和是←累加的也叫后缀和吧。正负分别表示某种果酱多出的数量,相加相等则表示数量相等。如上的红色字体位置,二者之间的果酱都需要清除掉,共6瓶。为了使清除的果酱数量尽可能少,相同的前后缀和,左边前缀和尽量往右靠,右边后缀和尽量往左靠。

又如另一组数据

原数据 1,0,0,      1,0,0

变为   1,-1,-1,    1,-1,-1

前缀和 1,0,-1,    -1,-2,-1

显然去掉左边2瓶就是答案。

选择红色位置用aid数组和bid数组,这两个数组起到指示下标的作用。

第三组数据的应对方法详看代码。

第四种数组1,2,  1,1或者1,1,  2,1之类的,需要对0做出处理。在这个数据上wa很多次。详看代码。

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner scan=new Scanner(System.in);
        int n,x;
        
        int  t=scan.nextInt();
        while(t!=0) {
            int p=100000;
            int [] a=new int[100005];//左边果酱前缀和 →
            int [] b=new int[100005];//右边果酱后缀和 ←
            int [] aid=new int[200005];
            int [] bid=new int[200005];
            //前缀和的数字下标,aid[2]=5表示a前缀数组内容为2的在前缀和数组下标为5的位置
            t--;
            n=scan.nextInt();
            aid[p]=0;
            for(int i=1;i<=n;i++) {//从左到右输入,直接计算前缀和
                x=scan.nextInt();
                if(x==1)
                    a[i]=a[i-1]+1;
                else 
                    a[i]=a[i-1]-1;
                
                int idx=a[i]+p;//防止遇到负数导致数组越界
                aid[ idx ]=i;//前缀和尽量往右靠,i越来越大
            }
            
            for(int i=1;i<=n;i++) {//从左到右输入
                x=scan.nextInt();
                if(x!=1)
                    x=-1;
                b[i]=x;
            }
            bid[p]=n+1;
            for(int i=n;i>=1;i--) {//从右往左计算后缀和
                b[i]+=b[i+1];
                
                int idx=b[i]+p;
                bid[idx]=i;//后缀和尽量往左靠,i越来越小
            }

            int ans=2*n;//最多把全部果酱拿走,应付第3组数据
            for(int i=-n;i<=n;i++) {
                if( (aid[p+i]!=0 && bid[p-i]!=0) || (i==0)) 
                    ans=Math.min(ans,n-aid[p+i]+bid[p-i]-1);
            }
            System.out.println(ans);
        }
    }

}

CF1278C-Berry Jam-(前缀和)

标签:指示   越界   http   def   target   public   内容   多次   +=   

原文地址:https://www.cnblogs.com/shoulinniao/p/12075697.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!