548. Split Array with Equal Sum

这道看上去不难,其实有不少坑。
笔者刚开始把打算用我最爱的双指针,L, R,如果L左边的sum 和 L右边的sum相等,那我在 L,R之间找一找能不能解决也找到两对和他们的sum相等的。 可提交之后发现坑了。
如果LR, 两sum不相等, 那怎么弄? 因为这些个数字可以是负数,所以即使如果 L这边的值大,也不一定要移动R指针。
。。。。所以笔者就陷入了两难。

然后就尝试DFS吧。列举各种可能,然后剪枝, 反正也就4层DFS。 其实也就三层。大量剪枝后效果也不是很差。下面是DFS的做法。这个最坏是ON^3的写法。实际运行起来没那么差因为剪枝
这个做法写的挺丑的 主要DFS分各种情况,不够美观 。

    public boolean splitArray(int[] nums) {
        int N = nums.length;
        Map<Integer, Set<Integer>> prefixMap = new HashMap<>();
        int[] prefixSum = new int[N];
        int preSum = 0;
        for (int i = 0; i < N; i++) {
            preSum += nums[i];
            prefixSum[i] = preSum;
            prefixMap.putIfAbsent(preSum, new HashSet<>());
            prefixMap.get(preSum).add(i);
        }
        
        return dfs(nums, 0, 0, 0, prefixSum, prefixMap);
    }
    private boolean dfs(int[] nums, int index, int depth, int sum, int[] prefixSum, Map<Integer, Set<Integer>> prefixMap) {
        if (index > nums.length - 1) return false;
        if (depth == 3) { //第四层,直接判断退出
            return (sum == prefixSum[prefixSum.length - 1] - prefixSum[index - 1]);
        }
        if (depth == 0) { // 第一层,枚举所有可能 
            for (int i = index; i < nums.length; i++) {
                if (i != index && nums[i] == 0) continue; // 对于第二个0就不要算了。
                if (depth == 0 || prefixSum[i] - prefixSum[index - 1] == sum) {
                    if (dfs(nums, i + 2, depth + 1, depth == 0 ? prefixSum[i] : sum,  prefixSum, prefixMap)) return true;
                }
            }
            return false;
        }
        // 第二层第三层, 利用prefixMap大量剪枝
        if (prefixMap.get(sum + prefixSum[index - 1]) == null) return false;
        for (int candidate : prefixMap.get(sum + prefixSum[index - 1]) ) {
            if (candidate >= index & candidate + 2 < nums.length) {
                if (dfs(nums, candidate + 2, depth + 1, sum, prefixSum, prefixMap)) return true;;
            }
        }
        return false;
    }

下面讨论下看来的别人的做法。
leetcode上高票答案是sweep中间那个档板,然后sweep左边看能不能分成相等的两半,如果能,就把它放在hashset里;然后sweep右边那一半,能不能分成相等两半,如果能看看和左边有没有正好相同的配对。这个想法很赞。时间复杂度来算一下, O (N * ( N + N) ) = O(N^2)

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

推荐阅读更多精彩内容

  • Given an array with n integers, you need to find if there...
    matrxyz阅读 904评论 0 0
  • 对于 D 题的原题意,出题人和验题人赛前都没有发现标算存在的问题,导致了许多选手的疑惑和时间的浪费,在此表示真诚的...
    _Carryon阅读 251评论 0 0
  • 上个月和亲戚一起去逛四川省博物馆,看到了很多书法作品,还有很多人在一个区域里面,用毛笔抄经书。在这样的场景下,诱发...
    猫女杜瑜阅读 491评论 0 1
  • “满纸荒唐言,一把辛酸泪。都云作者痴,谁解其中味?”曹雪芹的红楼梦写尽了人间的悲欢离合,让读者潸然泪下。 ...
    162LTT阅读 219评论 0 0
  • 流浪的未曾归 还是找不到家? 黑夜乍短还长 日间昼长愈短 除了细纹和白发 一如既往的自不可拔 如未想过执手天涯 怎...
    墙里格格巫阅读 80评论 0 0