傳送

題意：找出兩個序列，各刪除一個數後序列和相等，問刪除哪兩個序列的哪兩個數
題解：預處理出刪除每個數後的序列和，set存一下。

#include<bits/stdc++.h>
using namespace std;
int k,a,sum[200005];
vector<int> b[200005];
set<pair<int ,int> >s;
map<int,pair<int,int> >mp;
int main()
{
    cin>>k;
    for(int i=0;i<k;i++)
    {
        cin
 
>>a;
        for(int j=0;j<a;j++)
        {
            int d;
            cin>>d;
            b[i].push_back(d);
            sum[i]+=d;
        }

    }
    for(int i=0;i<k;i++)
    {
        for(int j=0;j<a;j++)
        {
            int t=sum[i]-b[i][j];
            if(mp.find(t)!=mp.end()&&mp[sum[i]-b[i][j]].first!=i+1
 
)
            {
                cout<<"YES"<<endl;
                cout<<i+1<<' '<<j+1<<endl;
                cout<<mp[sum[i]-b[i][j]].first<<' '<<mp[sum[i]-b[i][j]].second<<endl;
                return 0;
            }
            mp[sum[i]-b[i][j]]=make_pair(i+1
 
,j+1);
        }
    }
    cout<<"NO"<<endl;
}