codeforces#486(div3)c

人生的第一篇博客>_<(欢迎小伙伴来指教)
/*
补充一点基础知识——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");

}



阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭