题目链接:点击打开链接
题目大意:给出n个序列,然后求是否存在两个序列,如果他们各自减少一个数字,那么他们其余的数字和相等。
题目思路:我们通过一个数组来a来记录信息,ai=(sumi,seqi,eli)
其中sumi表示减去元素xi后的和,seq表述序列号,eli表示减去的元素的下标。
将所有信息存储在这个数组之后,按照sum和的大小排序即可,如果存在相邻的两个ai和ai+1的sum相等,但seq序列不等的号,那我们就找到我们所需要找的结果了。
#include<bits/stdc++.h>
using namespace std;
const double eps=1e-10;
const int maxn=1000+10;
int main()
{
int n,m;
cin>>n;
vector<pair<int,pair<int ,int > > > a;
for(int i=0;i<n;i++)
{
cin>>m;
vector<int>x(m);
int sum=0;
for(int j=0;j<m;j++)
{
cin>>x[j];
sum+=x[j];
}
for(int j=0;j<m;j++)
{
a.push_back(make_pair(sum-x[j],make_pair(i,j)));
}
}
sort(a.begin(),a.end());
for(int i=0;i<a.size()-1;i++)
{
if(a[i].first==a[i+1].first&&a[i].second.first!=a[i+1].second.first)
{
printf("YES\n");
printf("%d %d\n",a[i].second.first+1,a[i].second.second+1);
printf("%d %d\n",a[i+1].second.first+1,a[i+1].second.second+1);
return 0;
}
}
printf("NO\n");
return 0;
}