• 【LeetCode Weekly Contest 26 Q4】Split Array with Equal Sum


    【题目链接】:https://leetcode.com/contest/leetcode-weekly-contest-26/problems/split-array-with-equal-sum/

    【题意】

    让你把一段序列去掉3个元素,然后分成4个部分;
    要求这4个部分的和相同;
    问你可不可能;

    【题解】

    先枚举要删除的3个元素中的中间那个元素j;
    然后把整个序列分成左边和右边两个部分;
    然后再左边的序列中枚举i;
    然后把0..j-1分成0..i-1和i+1..j-1两个部分;
    然后看这两个部分各自的和是不是相同;
    相同的话把和用map记录一下;
    然后再枚举
    右半部分即第3个元素k;
    分成
    j+1..k-1和k+1..n-1两个部分;
    看看各自的和是不是相同;
    相同的话,看看刚才左半部分记录的map里面有没有这个和;
    有的话输出true
    复杂度接近O(n^2)吧->用了map哦。

    【完整代码】

    #include <bits/stdc++.h>
    using namespace std;
    #define lson l,m,rt<<1
    #define rson m+1,r,rt<<1|1
    #define LL long long
    #define rep1(i,a,b) for (int i = a;i <= b;i++)
    #define rep2(i,a,b) for (int i = a;i >= b;i--)
    #define mp make_pair
    #define ps push_back
    #define fi first
    #define se second
    #define rei(x) scanf("%d",&x)
    #define rel(x) scanf("%lld",&x)
    #define ref(x) scanf("%lf",&x)
    
    typedef pair<int, int> pii;
    typedef pair<LL, LL> pll;
    
    const int dx[9] = { 0,1,-1,0,0,-1,-1,1,1 };
    const int dy[9] = { 0,0,0,-1,1,-1,1,-1,1 };
    const double pi = acos(-1.0);
    const int N = 2e3+100;
    
    int pre[N],n;
    map <int, int> dic;
    
    class Solution {
    public:
        bool splitArray(vector<int>& nums) {
            n = nums.size();
            pre[0] = nums[0];
            rep1(i, 1, n - 1)
                pre[i] = pre[i - 1] + nums[i];
            rep1(j, 1, n - 2)
            {
                dic.clear();
                int l = 0, r = j - 1;
                rep1(i, l + 1, r - 1)
                {
                    if (pre[i] - nums[i] == pre[r] - pre[i])
                    {
                        dic[pre[r] - pre[i]] = 1;
                    }
                }
    
                l = j + 1, r = n - 1;
                rep1(i, l + 1, r - 1)
                {
                    if (dic[pre[r] - pre[i]] && (pre[i] - nums[i] - pre[j]) == (pre[r] - pre[i]))
                        return true;
                }
            }
            return false;
        }
    };
  • 相关阅读:
    python安装mysqldb
    2月8日
    python反射机制
    python备忘
    Nginx+Tomcat动静分离及Nginx优化
    yum挂在iso文件yum源配置
    升级apache
    解决编译apache出现的问题:configure: error: APR not found . Please read the documentation
    学习网站总结
    面试题:登录页面测试
  • 原文地址:https://www.cnblogs.com/AWCXV/p/7626496.html
Copyright © 2020-2023  润新知