紫书5-UVA-1592-数据库-map技巧
紫书5-UVA-1592-数据库-map技巧

题意:

给出一个表格,问表格中是否有某两行和某两列相等,即(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;
}

相关文章: