题意:
给出一个表格,问表格中是否有某两行和某两列相等,即(r1,c1)=(r2,c1),(r1,c2)=(r2,c2),并输出位置
思路:
若用四重循环枚举各行格列一定会超时,所以只枚举每两行,再从上到下枚举这两行的所有列,将每次的值用pair维护,再将整个pair存入map,映射到相应列数,
若map中存在相同pair,即证明当前两行中有答案,且两列数分别为map[pair],和当前列数k。
再考虑如何将两行的值储存在一起,因题中给出的值都是string类型,map内部实现时需要排序,对于string排序会耗费大量时间,应先将每种string映射到唯一int数,再用pair存储
代码:
#include <iostream>
#include <set>
#include <map>
#include <string>
using namespace std;
map<string,int>code;
map<pair<int,int>,int>dict;
int getid(string a)
{
if(code.count(a))
return code[a];
else
return code[a]=code.size();
}
int data[10005][15];
int main()
{
int n,m;
while(cin>>n>>m)
{
getchar();
code.clear();
dict.clear();
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
string a;
if(j<m-1)
getline(cin,a,',');
if(j==m-1)
getline(cin,a);
int id=getid(a);
data[i][j]=id;
}
}
bool flag=0;
int ans[2][2];
for(int i=0;i<m-1;i++)
{
for(int j=i+1;j<m;j++)
{
for(int k=0;k<n;k++)
{
int x=data[k][i];
int y=data[k][j];
if(dict.count(make_pair(x,y)))
{
flag=1;
ans[0][0]=dict[make_pair(x,y)];
ans[0][1]=k;
ans[1][0]=i;
ans[1][1]=j;
goto A;
}
else
dict[make_pair(x,y)]=k;
}
dict.clear(); //注意遍历完每两行后要清空map
}
}
A:
if(flag)
{
cout<<"NO"<<endl;
cout<<ans[0][0]+1<<" "<<ans[0][1]+1<<endl;
cout<<ans[1][0]+1<<" "<<ans[1][1]+1<<endl;
}
else
cout<<"YES"<<endl;
}
return 0;
}