未排序数组中累加和为给定值的最长子数组系列问题

题目1

给定一个无需数组arr,其中元素可正,可负,可0,给定一个整数k。求arr的所有子数组中累加和为k的最长子数组的长度。

思路

首先注意到这里元素可正可负可0,所以之前的两个指针的方法不再有效,这里需要使用一种全新的方法,联想到之前说过的数组问题可以考虑必须以j结尾的情况下怎么怎么样,所以这个题按这种思路进行下去。
假设我们能得到以必须以位置j结尾的满足条件的最长子序列,那么我们从这些最长中找到一个最大即为所求,那么怎么求解以位置j结尾的累加和为k的最长子序列呢?我们假设
sum[j]为从0累加到j的和,那么以位置j结尾的满足条件的最长子序列的开头就应该是从0开始第一次出现累加和为sum[j]-k的位置i,也就是sum[i]+target=sum[j],这里sum代表从开始的累加和。

算法流程

  1. 记录一个全局最大res,记录一个sum表示从0位置一直加到当前位置的累加和,使用一个map来记录前边遍历过程中出现过的累加和。
  2. 遍历数组,对于当前位置i,首先更新sum,然后看map中有没有sum-k这个键,如果有,则取出这个键对应的值j,表明(i,j]这一小段子序列的和就为k,用这个k去更新我们的res。遍历结束之后res就为我们的最大。如果没有,这把这个位置和sum放入map中。
  3. 需要特别注意的是首先要把(0,-1)放进map中。比如arr=[5,-1,-1,-1],k=5,如果不放入(0,-1),在考察位置0的时候,其实刚好这个和是满足条件的,但是被忽略了。

算法原理

这里的map记录的是累加和第一次出现某个值的位置。我们在遍历到j的时候,知道sum[j]=0,1,2,,,j的这些元素的累加和,sum[i]为0,1,,,,i这些数的累加和。考察位置j,我们需要找到第一次累加和为sum[j]-k的位置,如果找到了这个位置,那么就有i+1,i+2,...j这些元素的累加和为k,也就求得了以位置j结尾的累加和为k的最长子序列。我们遍历了数组一遍,所有位置结尾的,如果有满足条件的子序列,都求出来了,在这些中的最长的即为所求。

代码

  public static int maxLength(int[] arr, int k) {
        if(arr==null||arr.length==0)
            return 0;
        int sum=0;
        int res=0;
        Map<Integer,Integer> map=new HashMap<Integer,Integer>();
        map.put(0,-1);
        for(int i=0;i<arr.length;i++){
            sum+=arr[i];
            if(map.containsKey(sum-k)){
                res=Math.max(i-map.get(sum-k),res);
            }
            if(!map.containsKey(sum))
                map.put(sum,i);
        }
        return res;
    }

题目拓展

1. 给定一个无需数组arr,其中元素可正,可负,可0,求arr中所####有正数与负数个数相等的最长子数组长度。

可以这样想,原数组中正数就变为1,负数就变为-1,0就为0,另k等于0,调用上述方法即可。

2. 给定一个无需数组arr,其中元素只能为1或0,求arr所有子数组中0和1个数相等的最长子数组的长度。

1为1,0为-1,k为0,调用上述求解和为k的最长子数组算法即可。

题目2

未排序数组中,累加和小于或等于给定值的最长子数组长度。
例如:arr=[3,-2,-4,0,6],k=-2,则累加和小于或等于-2的最长子数组的长度为4。[3,-2,-4,0]

思路

和之前的区别是,这里累加和是小于等于k,那么借助于之前的思路,我们考虑以位置j结尾的累加和小于等于k的最长子数组的长度,我们自然可以想到,需要寻找第一个累加和大于等于k的位置(之前的思路是第一个等于k的位置,我们是借助于一个map来实现的)那么这里怎么办呢?由于我们只关心第一个大于某个值的位置,所以我们可以借助一个阶梯上升的辅助数组h来实现,h[i]表示从开始位置到当前位置的最大累加和。这样在寻找第一个大于某个值得位置时,我们只需要二分来搜索即可。

arr:  3  2  -3  2  -2  6  -3  1
sum:  3  5   2  4   2  8   5  6
h:    3  5   5  5   5  8   8  8 

算法流程&原理

使用一个辅助数组h,h[i]表示从开始到i位置的数组的子数组的最大和(不包括i位置)。然后遍历数组,二分查找第一个大于sum-k的位置,找到则更新,找不到继续。
之前一直不明白为什么h需要h[0]为0,然后长度为len+1,现在终于想明白了,因为我们求得第一个大于等于sum-k的位置,之前的方法h[i]包含了i位置的和,所以

3 2 -1 2 4  5  6 -1   5
3 5  4 6 10 15 21 20 25
3 5  5 6 10 15 21 21 25

比如说现在计算21这个位置,假设k为16,那么就需要求第一个大于等于5的位置i,并且这个位置我们是不能要的,这样i+1...j位置的和才会小于等于16。假设我们h[i]记录的就是0...i的这个递增的数组的话,比如上边h[1]表示加到h[1]最大和为5,我们不应该要h[1]而应该从h[2]开始,但是假设某些情况下h[2]也不满足条件,这就很麻烦了(3 2 0 2 4 5 6 -1 5),所以我们让h[i]为[0,i)的,不包含位置i,我们找到一个位置后,可以知道这个位置的前面是大于等于5的,这个位置就一定是满足条件的。

代码

    public static int maxLength(int[] arr, int k) {
        if(arr==null||arr.length==0)
            return 0;
        int[] h=new int[arr.length+1];
        h[0]=0;
        int sum=arr[0];
        for(int i=1;i<arr.length;i++){
            sum+=arr[i];
            h[i+1]=Math.max(sum,h[i]);
        }
        int res=0;
        int len;
        int index;
        sum=0;
        for(int i=0;i<arr.length;i++){
            sum+=arr[i];
            index=firstBiggerOrEqual(h,sum-k);
            len=index==-1?0:i-index+1;
            res=Math.max(res,len);
        }
        return res;
    }
    public static int firstBiggerOrEqual(int[] h,int num){
        int left=0;
        int right=h.length-1;
        int res=-1;
        while(left<=right){
            int mid=(left+right)/2;
            if(h[mid]>=num){
                res=mid;
                right=mid-1;
            }
            else{
                left=mid+1;
            }
        }
        return res;
    }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,456评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,370评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,337评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,583评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,596评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,572评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,936评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,595评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,850评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,601评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,685评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,371评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,951评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,934评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,167评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,636评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,411评论 2 342

推荐阅读更多精彩内容