ACWing 3956.截断数组

题目

分类应做到不重不漏 通过闫式集合分析法来进行

y总讲解：

集合：对于第二刀切法位置的集合 即：从第几个数开始切第二刀
不能为空，所以从第二处开始，在n - 1处截止

一个暴力O(n^2)，正解O(n)

暴力 TLE 过了14/22个数据

#include <bits/stdc++.h>

using namespace std;

const int N = 100010;

int n,a[N],s[N];

int main()
{
    cin >> n;
    
    for(int i = 1;i <= n;i++)
    {
        cin >> a[i];
        s[i] = s[i - 1] + a[i];
    }
    
    int res = 0;
    for(int i = 1;i <= n - 2;i++)
      for(int j = 2;j <= n - 1;j++)
      {
          if(i < j)
          {
              if(s[i] == s[j] - s[i] && s[j] - s[i] == s[n] - s[j]) res++;
          }
      }
    
    cout << res << endl;
    
    return 0;
}

正解 O(n)

图例：对于第二刀 即 i = 3 与 第一刀

分清楚 对于 第一刀、第二刀 切之后 ，所分成的哪三个区间 ，并注意下标的表示

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
const int N = 100010;

int n,a[N],s[N];

int main()
{
    cin >> n;
    
    for(int i = 1;i <= n;i++)
    {
        cin >> a[i];
        s[i] = s[i - 1] + a[i];
    }
    
    if(s[n] % 3 != 0) 
    {
        puts("0");
        return 0; // 结束main函数的写法  也可用 else
    }
    
    LL res = 0,cnt = 0; // 组合数C(n-1) 2  n为10的5次方  所以用 long long 
    
    for(int i = 2;i <= n - 1;i++) // 对应上面的集合 
    {
        if(s[i - 1] == s[n] / 3) cnt++;         // 记录第二刀之前可以切多少刀  理清 第一刀 用 s[i - 1] 所表示的区间 
        if(s[n] - s[i] == s[n] / 3) res += cnt; // 第二刀 s[n] - s[i] 是第三个相等区间       // 也可以 s[i] == s[n] / 3 * 2
    }
    
    cout << res << endl;
    
    return 0;
}