2022-05-08每日刷题打卡

最新推荐文章于 2024-10-06 14:10:45 发布

你好_Ä

最新推荐文章于 2024-10-06 14:10:45 发布

阅读量438

点赞数

分类专栏：笔记文章标签：算法 c++ 图论

本文链接：https://blog.csdn.net/fnmdpnmsl/article/details/124654147

版权

笔记专栏收录该内容

160 篇文章 13 订阅

订阅专栏

该博客讨论了一种算法问题，即如何将一个给定的序列分割成三个连续子序列，使得每个子序列的元素和相等。通过计算序列的前缀和，并使用哈希映射存储特定和的索引，可以找到满足条件的分割方法。当序列总和不能被3整除时，无法进行分割。博客提供了详细的代码实现和样例解释。

摘要由CSDN通过智能技术生成

2022-05-08每日刷题打卡

代码源——每日一题

三段式 - 题目 - Daimayuan Online Judge

有一个长度为n的序列，现在我们想把它切割成三段(每一段都是连续的），使得每一段的元素总和都相同，请问有多少种不同的切割方法

输入描述

第一行给出一个数n，(1≤n≤10^5)

第二行给出序列a1，a2，a3，…，an，(|ai|≤10^5)

输出描述

输出一个数表示有多少种不同的切割方法

样例输入

4
1 2 3 3

样例输出

样例解释

可以将它分成第一组1，2，第二组3，第三组3

问题解析

前缀和，我们既然要把数组分成三组，那么实际上我们已经可以知道这三组的和是多少了，毕竟原数组的和是固定的，设为sum，现在我们要把他们分成三个子数组，且这三个子数组和要相等，相当于这三个子数组的和都是sum/3。所以我们可以先判断一下sum能否被3整除，要是不行，那么就肯定不能切割了。

我们可以先算出原数组的前缀和数组，然后根据sum找点分割数组，找到sum/3的点和sum/3*2的点，我们可以把这些点的下标分别存起来，然后在sum/3 *2的点前面找sum/3的点，算组合数。

#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>
#include<math.h>
#include<set>
#include<numeric>
#include<string>
#include<string.h>
#include<iterator>
#include<map>
#include<unordered_map>
#include<stack>
#include<list>
#include<queue>
#include<iomanip>

#define endl '\n';
typedef long long ll;
typedef pair<ll, ll>PII;
const int N = 1e4 + 50;

int main()
{
    int n;
    cin >> n;
    if (n < 3)
    {
        cout << 0 << endl;
        return 0;
    }
    vector<ll>v(n), sum(n);
    unordered_map<int, int>mymap;
    cin >> v[0];
    sum[0] = v[0];

    for (int i = 1; i < n; i++)cin >> v[i], sum[i] = sum[i - 1] + v[i];
    if (sum[n - 1] % 3 != 0)
    {
        cout << 0 << endl;
        return 0;
    }
    ll a = sum[n - 1] / 3, b = a * 2;
    set<ll>one, two;
    int cnt = 0;
    bool flag = true;
    for (int i = 1; i < n-1; i++)
    {
        if (sum[i] == a)
        {
            cnt++;
            one.insert(i);
            mymap[i] = cnt;
        }
        if (sum[i] == b)two.insert(i);
    }

    if (one.empty() || two.empty())
    {
        cout << 0 << endl;
        return 0;
    }
    ll res = 0;
   
    for (auto i : two)
    {

        auto it = one.upper_bound(i);
        it--;
        res += mymap[*it];
        

    }
    cout << res << endl;
    return 0;
}