leetcode1013:将数组分成和相等的三个部分

刚看题目就感觉不会做了~然后洗澡的时候想了想,这不挺简单的吗。

思路:定义一个vector,存入当前数值和之前数值之和,然后取最后一个数除以3 = a,再判断该vector里有没有a和2a,且a的索引必须小于2a

bool canThreePartsEqualSum(vector<int>& A) {
    int sum = 0;
    int fen;
    int flag = 0;
    //先遍历一遍,把当前数和之前数的和 赋值到vector A中,不用重新定义vector,节省资源
    for (int i = 0; i < A.size(); i++) {
        sum += A[i];
        A[i] = sum;
    }
    // 最后一个数除以3得到三分值
    fen = A[A.size()-1]/3;
    //再遍历一遍,判断vector中是否有三分值、三分值的两倍
    //有的话就返回true
    for (int j = 0; j < A.size()-1; j++)
    {
        if(A[j] == fen && flag == 0)
        {flag++;continue;}
        if(A[j] == 2*fen && flag == 1)
        {return true;}
    }
    return false;
}
发布了39 篇原创文章 · 获赞 50 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/Hanghang_/article/details/104807053
今日推荐