Educational Codeforces Round 157 1895C- Torn Lucky Ticket (div2 前缀和/思维 补)

题目链接： https://codeforces.com/contest/1895/problem/C

题目大意：

门票是从 $1$ 到 $9$ 的非空数字字符串 $s_i(1\le s_{i}length\le 5)$。
定义幸运门票为：

• 长度均匀
• 前半部分的数字之和等于后半部分的数字之和。

您将获得 $n$ 张门票 $s_1,s_2,s_3,\dots ,s_n$。请问有多少对 $(i,j)$ $(1\le i,j\le n)$ 使得 $s_i+s_j(eg.s_i=13,s_j=37,s_i+s_j=1337)$ 是一张幸运门票？请注意可能 $(i==j)$ 。

题目思路：

首先注意到 $n\le 2\ast 10^5$ 普通暴力枚举 $O(n^2)$ 肯定超时。不难发现只要利用前缀和记录 $s_i$ 到每个位置的数字之和，然后记录 $cnt[s_{i}length][s_{i}sum]$ ，然后枚举每个 $s_i$ 从 $1$ 到 $i(1\le i\le s_{i}length)$ 和 从 $i$ 到 $s_{i}length$ 的数字之和 所对应需要的 $cnt[length][sum]$ 是否存在。其中 $(length=2\ast s_{j}length-s_{i}length,sum=2\ast s_{j}sum-s_{i}sum)$ 。此时 时间复杂度为 $O(n)$ 。

题目代码：

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
const int N = 2e5+10;
int n, ans = 0;
int sum[N][7], cnt[10][N];
void solve()
{
    cin >> n;
    for (int i = 1; i <= n; i++) {
        string s;
        cin >> s;
        int len = s.size();
        for (int j = 0; j < len; j++) {
            sum[i][j + 1] = sum[i][j] + s[j] - '0';
        }
        sum[i][0] = len;
        int total = sum[i][len];
        cnt[len][total]++;
    }
    ans = 0;
    for (int i = 1; i <= n; i++) {
        int len = sum[i][0];
        int total = sum[i][len];
        ans += cnt[len][total];
        for (int j = 1; j < len; j++) {
            if (2 * j <= len)
                continue;
            ans += cnt[2 * j - len][2 * sum[i][j] - total];
            ans += cnt[2 * j - len][2 * (sum[i][len] - sum[i][len - j]) - total];
        }
    }
    cout << ans << endl;
}
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    // cout.tie(0);
    // int _;cin>>_;
    // while(_--)
    solve();
    return 0;
}