人生的第一篇博客>_<(欢迎小伙伴来指教)
/*
补充一点基础知识——c++的一个好玩的东西map,
使用count()函数,返回的是被查找元素的个数。有的话,返回1,否则为0;
使用find()函数,返回的是被查找元素的位置,没有的话返回map.end().
*/
这道题并不难,主要意思是要从给定的k个数组中,任意选出两个,然后对每个选定的数组删除一个数,执行这波骚操作后,两个数组的元素和相等。
提供一个合理的思路哈——对每个数组,我们记录删除任意一个元素后的元素和,之后输入下一列数组时,通过map映射,寻找前面的数组是否有同样的元素和,有的话,赶紧输出YES喽,否则为NO,具体实现前看下面的代码~~
#include<bits/stdc++.h>
using namespace std;
const int maxn = 2*1e5 + 5;
int a[maxn];
int main()
{
map<int , pair<int , int> > mp;
int T;
scanf("%d", &T);
for(int j = 0; j<T ; j++)
{
int num, sum = 0;
scanf("%d" , &num);
for(int i=0; i<num; i++)
{
scanf("%d", &a[i]);
sum += a[i];
}
for(int i=0; i<num; i++)
{
if(mp.count(sum-a[i]))
{
puts("YES");
printf("%d %d\n%d %d\n", mp[sum-a[i]].first, mp[sum-a[i]].second, j+1, i+1 );
//j 和 i 加1 是因为列数和下标都是从1开始计数,而代码中是从0开始,所以要人为加1
return 0;
}
}
for(int i=0; i<num; i++)
mp[sum-a[i]] = {j+1, i+1};//标记上是第几列 和 所在列数的下标
}
puts("NO");
}