【每日一题】CF988 C. Equal Sums | 排序

本文链接：https://blog.csdn.net/weixin_43900869/article/details/137776051

文章描述了一种算法，用于解决给定多个数组的问题，判断是否存在两个数组，通过删除一个元素使它们的和相等。通过排序和比较元素来确定答案，时间复杂度为O(nlogn)。

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

题目内容

原题链接

给定 $k$ 个数组，第 $i$ 个数组的长度为 $n_i$ 。

问是否可以选择两个数组，编号分别为 $a$ 和 $b$ ，从 $a$ 和 $b$ 中各删除一个数，使得两者的数组和相同。

如果可以，输出三行

第一行输出 YES
第二行输出选择的数组 $a$ 的编号以及数组 $a$ 中删除的数组元素的下标（从 $1$ 开始）
第三行输出选择的数组 $b$ 的编号以及数组 $b$ 中删除的数组元素的下标（从 $1$ 开始）

如果不可以，输出一行 NO

数据范围

$2\leq k\leq 2\cdot 10^5$
$1\leq n_i< 2\cdot 10^5$
$-10^4\leq a{i,j}\leq 10^4$ ， $a_{i,j}$ 为第 $i$ 个数组的第 $j$ 个元素
$\sum\limits_{i=1}^kn_i\leq 2\cdot 10^5$

题解

考虑对于第 $i$ 个数组，其所有元素和为 $s u m$ 。
将第 $i$ 个数组的第 $j$ 个元素删除后，存下 $sum-a_{i,j},i,j)$ 三元组

对于所有数组的所有元素，共有 $\sum\limits_{i=1}^kn_i$ 个元素

按照 $sum-a_{i,j}$ 排序这些三元组，然后判断第一关键字相同的三元组中，是否存在有 $i$ 不同的三元组，如果有则是存在答案，输出即可。

时间复杂度： $O(n\log n), n=\sum\limits_{i=1}^kn_i$

代码

#include <bits/stdc++.h>
using namespace std;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int k;
    cin >> k;

    vector<pair<int, pair<int, int>>> vec;
    for (int i = 1; i <= k; ++i) {
        int n;
        cin >> n;

        int sum = 0;
        vector<int> a(n);
        for (int j = 0; j < n; ++j) {
            cin >> a[j];
            sum += a[j];
        }

        for (int j = 0; j < n; ++j) {
            vec.emplace_back(sum - a[j], make_pair(i, j + 1));
        }
    }

    sort(vec.begin(), vec.end());
    int n = vec.size();

    for (int i = 0; i < n; ++i) {
        int j = i + 1;
        while (j < n && vec[j].first == vec[i].first) j += 1;
        if (vec[i].second.first != vec[j - 1].second.first) {
            cout << "YES\n";
            cout << vec[i].second.first << " " << vec[i].second.second << "\n";
            cout << vec[j - 1].second.first << " " << vec[j - 1].second.second << "\n";
            return 0;
        }

        i = j - 1;
    }

    cout << "NO\n";

    return 0;
}