PAT 1139 First Contact

题目链接:点击打开链接

思路:分析可知,有以下几个问题要处理,一是如何存储整个关系网;二是如何快速判断两个人之间是否存在朋友关系;三是如何将每个询问相应的结果存储并排序。题目没什么难度,采取合适的数据结构去处理以上三个问题即可,就是有一些坑点要注意。

七种方法(优化或不同方法):

一:前向星存图;利用set<pair<int,int> >快速判断朋友关系;set<pair<int,int>,cmp>存储结果

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <iostream>
#include <utility>
#include <set>
using namespace std;
struct node{
  int v;
  int next;
}edge[200000];//开小了出现段错误 
int cnt,head[10000];
int sex[10000];
void Add(int u,int v){
  cnt++;
  edge[cnt].v = v;
  edge[cnt].next = head[u];
  head[u] = cnt;
  return;
}
void Nint(){
  cnt = 0;
  memset(head,0,sizeof(head));
  memset(sex,-1,sizeof(sex));
  return;
}
int Change(char s[]){
  int num = 0;
  if(s[0] == '-'){
    for(int i = 1;i <= 4;i++){
      num = num*10 + s[i] - '0';
    }
    sex[num] = 0;
    return num;
  }
  else{
    for(int i = 0;i < 4;i++){
      num = num*10 + s[i] - '0';
    }
    sex[num] = 1;
    return num;
  }
}
struct cmp{
  bool operator()(const pair<int,int> &a,const pair<int,int> &b)const{
    if(a.first != b.first){
      return a.first < b.first;
    }
    else{
      return a.second < b.second; 
    }
  }
};
set<pair<int,int> > s;
set<pair<int,int>,cmp> re;
int main(){
  int n,m,k;
  char a[10],b[10];
  scanf("%d%d",&n,&m);
  Nint();
  for(int i = 0;i < m;i++){
    scanf("%s%s",a,b);//这样读取的原因是有易错点:-0000 
    int aa = Change(a);
    int bb = Change(b);
    Add(aa,bb);
    Add(bb,aa);
    s.insert(pair<int,int>(aa,bb));//使用make_pair会在PAT上编译错误 
    s.insert(pair<int,int>(bb,aa));
  }
  scanf("%d",&k);
  while(k--){
    re.clear();
    scanf("%s%s",a,b);
    int aa = Change(a);
    int bb = Change(b);
    for(int i = head[aa];i;i = edge[i].next){
      int c = edge[i].v;
      if(bb == c || sex[aa] != sex[c]) continue;//易错点,别忘记判断找到的B不是D 
      for(int j = head[bb];j;j = edge[j].next){
        int d = edge[j].v;
        if(aa == d || sex[bb] != sex[d]) continue;//易错点,别忘记判断找到的C不是A
        if(s.find(pair<int,int>(c,d)) != s.end()){
          re.insert(pair<int,int>(c,d));
        }
      }
    }
    printf("%d\n",re.size());
    for(set<pair<int,int>,cmp>::iterator it = re.begin();it != re.end();it++){
      printf("%04d %04d\n",it->first,it->second);//%04d为易错点 
    }
  }
  return 0;
}
/*
  pair<int,int> a(1,2);
  pair<int,int> c;
  c = make_pair<int,int>(2,3);//utility头文件中 
*/

二:vector存图;利用set<pair<int,int> >快速判断朋友关系;set<pair<int,int>,cmp>存储结果

优化:vector只存储和自己性别相同的朋友关系

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <iostream>
#include <utility>
#include <set>
#include <vector>
using namespace std;
vector<int> v[10000];
int sex[10000];
int Change(char s[]){
	int num = 0;
	if(s[0] == '-'){
		for(int i = 1;i <= 4;i++){
			num = num*10 + s[i] - '0';
		}
		sex[num] = 0;
		return num;
	}
	else{
		for(int i = 0;i < 4;i++){
			num = num*10 + s[i] - '0';
		}
		sex[num] = 1;
		return num;
	}
}
struct cmp{
	bool operator()(const pair<int,int> &a,const pair<int,int> &b)const{
		if(a.first != b.first){
			return a.first < b.first;
		}
		else{
			return a.second < b.second; 
		}
	}
};
set<pair<int,int> > s;
set<pair<int,int>,cmp> re;
int main(){
	int n,m,k;
	char a[10],b[10];
	scanf("%d%d",&n,&m);
	for(int i = 0;i < m;i++){
		scanf("%s%s",a,b);//这样读取的原因是有易错点:-0000 
		int aa = Change(a);
		int bb = Change(b);
		if(sex[aa] == sex[bb]){//只存和自己性别相同的朋友 
			v[aa].push_back(bb);
			v[bb].push_back(aa);
		}
		s.insert(pair<int,int>(aa,bb));//使用make_pair会在PAT上编译错误 
		s.insert(pair<int,int>(bb,aa));
	}
	scanf("%d",&k);
	while(k--){
		re.clear();
		scanf("%s%s",a,b);
		int aa = Change(a);
		int bb = Change(b);
		for(int i = 0;i < v[aa].size();i++){
			int c = v[aa][i];
			if(bb == c || sex[aa] != sex[c]) continue;//易错点,别忘记判断找到的B不能是D 
			for(int j = 0;j < v[bb].size();j++){
				int d = v[bb][j];
				if(aa == d || sex[bb] != sex[d]) continue;//易错点,别忘记判断找到的C不能是A
				if(s.find(pair<int,int>(c,d)) != s.end()){
					re.insert(pair<int,int>(c,d));
				}
			}
		}
		printf("%d\n",re.size());
		for(set<pair<int,int>,cmp>::iterator it = re.begin();it != re.end();it++){
			printf("%04d %04d\n",it->first,it->second);//%04d为易错点 
		}
	}
	return 0;
}

三:vector存图;利用set<pair<int,int> >快速判断朋友关系;set<pair<int,int>,cmp>存储结果

优化:读取ID时,直接用string读取,利用string的length就可判断出性别是否一样,然后vector也只是存储性别相同的朋友关系。把string转化为无符号整数直接利用stoi(c++中)和abs(C语言中)函数即可。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iostream>
#include <utility>
#include <set>
#include <vector>
using namespace std;
vector<int> v[10000];
struct cmp{
	bool operator()(const pair<int,int> &a,const pair<int,int> &b)const{
		if(a.first != b.first){
			return a.first < b.first;
		}
		else{
			return a.second < b.second; 
		}
	}
};
set<pair<int,int> > s;
set<pair<int,int>,cmp> re;
int main(){
	int n,m,k;
	string a,b;
	scanf("%d%d",&n,&m);
	for(int i = 0;i < m;i++){
		cin >> a >> b;//这样读取的原因是有易错点:-0000
		int aa = abs(stoi(a));//stoi是C++的string中的函数;与atoi不同,atoi是C语言中的 
		int bb = abs(stoi(b));//abs也和fabs一样,是C语言math.h中的函数 
		if(a.length() == b.length()){
			v[aa].push_back(bb);
			v[bb].push_back(aa);
		}
		s.insert(pair<int,int>(aa,bb));//使用make_pair会在PAT上编译错误
		s.insert(pair<int,int>(bb,aa));
	}
	scanf("%d",&k);
	while(k--){
		re.clear();
		int aa,bb;
		scanf("%d%d",&aa,&bb);
		aa = abs(aa);
		bb = abs(bb);
		for(int i = 0;i < v[aa].size();i++){
			int c = v[aa][i];
			if(bb == c) continue;//易错点,别忘记判断找到的B不能是D 
			for(int j = 0;j < v[bb].size();j++){
				int d = v[bb][j];
				if(aa == d) continue;//易错点,别忘记判断找到的C不能是A
				if(s.find(pair<int,int>(c,d)) != s.end()){
					re.insert(pair<int,int>(c,d));
				}
			}
		}
		printf("%d\n",re.size());
		for(set<pair<int,int>,cmp>::iterator it = re.begin();it != re.end();it++){
			printf("%04d %04d\n",it->first,it->second);//%04d为易错点 
		}
	}
	return 0;
}

四:vector存图;利用set<pair<int,int> >快速判断朋友关系;vector<node>存储结果

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iostream>
#include <utility>
#include <set>
#include <vector>
using namespace std;
vector<int> v[10000];
set<pair<int,int> > s;
struct node{
	int aa,bb;
	node(int c,int d):aa(c),bb(d){}
	bool operator<(const node &c)const{
		if(this->aa != c.aa){
			return this->aa < c.aa; 
		} 
		else{
			return this->bb < c.bb;
		}
	}
};
vector<node> re;
int main(){
	int n,m,k;
	string a,b;
	scanf("%d%d",&n,&m);
	for(int i = 0;i < m;i++){
		cin >> a >> b;//这样读取的原因是有易错点:-0000
		int aa = abs(stoi(a));//stoi是C++的string中的函数;与atoi不同,atoi是C语言中的 
		int bb = abs(stoi(b));//abs也和fabs一样,是C语言math.h中的函数 
		if(a.length() == b.length()){
			v[aa].push_back(bb);
			v[bb].push_back(aa);
		}
		s.insert(pair<int,int>(aa,bb));//使用make_pair会在PAT上编译错误
		s.insert(pair<int,int>(bb,aa));
	}
	scanf("%d",&k);
	while(k--){
		re.clear();
		int aa,bb;
		scanf("%d%d",&aa,&bb);
		aa = abs(aa);
		bb = abs(bb);
		for(int i = 0;i < v[aa].size();i++){
			int c = v[aa][i];
			if(bb == c) continue;//易错点,别忘记判断找到的B不能是D 
			for(int j = 0;j < v[bb].size();j++){
				int d = v[bb][j];
				if(aa == d) continue;//易错点,别忘记判断找到的C不能是A
				if(s.find(pair<int,int>(c,d)) != s.end()){
					re.push_back(node(c,d));
				}
			}
		}
		sort(re.begin(),re.end());
		printf("%d\n",re.size());
		for(int i = 0;i < re.size();i++){
			printf("%04d %04d\n",re[i].aa,re[i].bb);//%04d为易错点 
		}
	}
	return 0;
}

五:vector存图;利用set<pair<int,int> >快速判断朋友关系;vector<pair<int,int> >存储结果

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iostream>
#include <utility>
#include <set>
#include <vector>
using namespace std;
vector<int> v[10000];
bool cmp(pair<int,int> a,pair<int,int> b){
	if(a.first != b.first){
		return a.first < b.first;
	}
	else{
		return a.second < b.second; 
	}
}
set<pair<int,int> > s;
vector<pair<int,int> > re;
int main(){
	int n,m,k;
	string a,b;
	scanf("%d%d",&n,&m);
	for(int i = 0;i < m;i++){
		cin >> a >> b;//这样读取的原因是有易错点:-0000
		int aa = abs(stoi(a));//stoi是C++的string中的函数;与atoi不同,atoi是C语言中的 
		int bb = abs(stoi(b));//abs也和fabs一样,是C语言math.h中的函数 
		if(a.length() == b.length()){
			v[aa].push_back(bb);
			v[bb].push_back(aa);
		}
		s.insert(pair<int,int>(aa,bb));//使用make_pair会在PAT上编译错误
		s.insert(pair<int,int>(bb,aa));
	}
	scanf("%d",&k);
	while(k--){
		re.clear();
		int aa,bb;
		scanf("%d%d",&aa,&bb);
		aa = abs(aa);
		bb = abs(bb);
		for(int i = 0;i < v[aa].size();i++){
			int c = v[aa][i];
			if(bb == c) continue;//易错点,别忘记判断找到的B不能是D 
			for(int j = 0;j < v[bb].size();j++){
				int d = v[bb][j];
				if(aa == d) continue;//易错点,别忘记判断找到的C不能是A
				if(s.find(pair<int,int>(c,d)) != s.end()){
					re.push_back(pair<int,int>(c,d));
				}
			}
		}
		sort(re.begin(),re.end(),cmp);
		printf("%d\n",re.size());
		for(int i = 0;i < re.size();i++){
			printf("%04d %04d\n",re[i].first,re[i].second);//%04d为易错点 
		}
	}
	return 0;
}

六:vector存图;利用二维数组快速判断朋友关系;vector<pair<int,int> >存储结果

优化;直接利用二维数组标记朋友关系,方便。之前感觉就算开为bool,65536kB的内存还是不够bool[10^4][10^4],但试了一下可以,应该和系统内存分配有关,但要注意的一点是,当使用memset,会超出内存限制。即使这样,超过内存限制或者超过语言限制的过大数组最好不要使用!!!

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iostream>
#include <utility>
#include <set>
#include <vector>
using namespace std;
vector<int> v[10000];
bool map[10000][10000];
bool cmp(pair<int,int> a,pair<int,int> b){
	if(a.first != b.first){
		return a.first < b.first;
	}
	else{
		return a.second < b.second; 
	}
}
vector<pair<int,int> > re;
int main(){
	int n,m,k;
	string a,b;
	scanf("%d%d",&n,&m);
	//memset(map,false,sizeof(map));加了这个超出内存限制 
	for(int i = 0;i < m;i++){
		cin >> a >> b;//这样读取的原因是有易错点:-0000
		int aa = abs(stoi(a));//stoi是C++的string中的函数;与atoi不同,atoi是C语言中的 
		int bb = abs(stoi(b));//abs也和fabs一样,是C语言math.h中的函数 
		if(a.length() == b.length()){
			v[aa].push_back(bb);
			v[bb].push_back(aa);
		}
		map[aa][bb] = true;
		map[bb][aa] = true;
	}
	scanf("%d",&k);
	while(k--){
		re.clear();
		int aa,bb;
		scanf("%d%d",&aa,&bb);
		aa = abs(aa);
		bb = abs(bb);
		for(int i = 0;i < v[aa].size();i++){
			int c = v[aa][i];
			if(bb == c) continue;//易错点,别忘记判断找到的B不能是D 
			for(int j = 0;j < v[bb].size();j++){
				int d = v[bb][j];
				if(aa == d) continue;//易错点,别忘记判断找到的C不能是A
				if(map[c][d]){
					re.push_back(pair<int,int>(c,d));
				}
			}
		}
		sort(re.begin(),re.end(),cmp);
		printf("%d\n",re.size());
		for(int i = 0;i < re.size();i++){
			printf("%04d %04d\n",re[i].first,re[i].second);//%04d为易错点 
		}
	}
	return 0;
}

七:vector存图;利用二维数组快速判断朋友关系;vector<node>存储结果

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
struct node{
	int aa,bb;
	node(int a,int b):aa(a),bb(b){}
	bool operator<(const node &a)const{
		return aa != a.aa?aa < a.aa:bb < a.bb; 
	} 
};
bool map[10000][10000];
int main(){
	int n,m,k;
	string a,b;
	vector<int> v[10000];
	//memset(map,false,sizeof(map));加了这个超出内存限制 
	scanf("%d%d",&n,&m);
	for(int i = 0;i < m;i++){
		cin >> a >> b;//这样读取的原因是有易错点:-0000
		int aa = abs(stoi(a));//stoi是C++的string中的函数;与atoi不同,atoi是C语言中的 
		int bb = abs(stoi(b));//abs也和fabs一样,是C语言math.h中的函数 
		if(a.length() == b.length()){
			v[aa].push_back(bb);
			v[bb].push_back(aa);
		}
		map[aa][bb] = true;
		map[bb][aa] = true;
	}
	scanf("%d",&k);
	while(k--){
		int aa,bb;
		vector<node> re;
		scanf("%d%d",&aa,&bb);
		aa = abs(aa);
		bb = abs(bb);
		for(int i = 0;i < v[aa].size();i++){
			int cc = v[aa][i];
			if(bb == cc) continue;//易错点,别忘记判断找到的B不能是D
			for(int j = 0;j < v[bb].size();j++){
				int dd = v[bb][j];
				if(aa == dd) continue;//易错点,别忘记判断找到的C不能是A
				if(map[cc][dd]){
					re.push_back(node(cc,dd));//node{cc,dd}也可以,C++ 11新标准
				}
			}
		}
		sort(re.begin(),re.end());
		printf("%d\n",re.size());
		for(int i = 0;i < re.size();i++){
			printf("%04d %04d\n",re[i].aa,re[i].bb);//%04d为易错点 
		}
	}
	return 0;
}


  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值