【网易编程题】圆环切割

题目:
小易有n个数字排成一个环,你能否将它们分成连续的两个部分(即在环上必须连续),使得两部分的和相等?


思路:
开始时,将第1个元素当作序列1,其余元素组成序列2。i1指向序列1的起始位置,i2指向序列2的起始位置。当序列1之和小于序列2之和时,i1向前移动,吃掉序列2的尾巴;反之,i2向前移动,序列2吃掉序列1的尾巴。如果序列1之和与序列2之和相等,则成功。否则,如果i2循环一圈又回到了初始位置,则失败。

数据结构:
vector模拟圆环,通过 (vec.size()+i+1)%vec.size() 来使i指向圆环的下一个元素。


#include 
#include 
#include 
using namespace std;
int cmp(const long long &a, const long long &b){
    return a > b;
}
int main(){
    int total;
    cin >> total;
    
    for (int i = 0; i < total; ++i){
        int n;
        cin >> n;
        vector<long long> vec;
        long long sum = 0;
        for (int j = 0; j < n; ++j){
            int t;
            cin >> t;
            vec.push_back(t);
            sum += vec[j];
        }
        long long sum1 = vec[0];
        long long sum2 = sum - vec[0];
        int i1 = 0, i2 = vec.size()-1;
        bool next = false;
        while (true){
            if (sum1 < sum2){
                i1 = (vec.size()+i1+1)%vec.size();
                sum1 += vec[i1];
                sum2 -= vec[i1];
            }else if(sum1 > sum2){
                i2 = (vec.size()+i2+1)%vec.size();
                sum1 -= vec[i2];
                sum2 += vec[i2];
            }else{
                break;
            }
            if (i2 == vec.size()-1){
                if (next){
                    break;
                }
            }else{
                next = true;
            }
        }
        
        if (sum1 == sum2){
            cout << "YES"<< endl;
        }else{
            cout << "NO" << endl;
        }
    }
}

你可能感兴趣的:(算法杂烩)