codeforces 988C

 C. Equal Sums

time limit per test：2 seconds    memory limit per test：256 megabytes

https://codeforces.com/problemset/problem/988/C

 

神奇的STL库。

 

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

vector<pair<int, pair<int, int> > > input;

int main()
{
    int k;
    cin>>k;
    for(int i=1;i<=k;i++){
        int n;
        cin>>n;
        vector<long long> temp(n+1); //如果想直接输入某一位置，必须规定初始大小。否则size为0啊
        long long sum=0;
        for(int j=1;j<=n;j++){
            cin>>temp[j];
            sum+=temp[j];
        }
        for(int j=1;j<=n;j++){
            input.push_back(make_pair(sum-temp[j], make_pair(i,j)));
        }
    }
    sort(input.begin(),input.end()); //pari可以直接比较大小，先比first再比second
    //input.size()返回的是无符号整数
    for(int i=0;i<int(input.size())-1;i++){
        if(input[i].first==input[i+1].first && input[i].second.first!=input[i+1].second.first){
            cout<<"YES"<<endl;
            cout<<input[i].second.first<<" "<<input[i].second.second<<endl;
            cout<<input[i+1].second.first<<" "<<input[i+1].second.second<<endl;
            return 0;
        }
    }
    cout<<"NO"<<endl;
    return 0;
}