打卡信奥刷题（1146）用C++实现信奥 P2078 朋友

P2078 朋友

题目背景

小明在 A 公司工作，小红在 B 公司工作。

题目描述

这两个公司的员工有一个特点：一个公司的员工都是同性。

A 公司有 $N$ 名员工，其中有 $P$ 对朋友关系。B 公司有 $M$ 名员工，其中有 $Q$ 对朋友关系。朋友的朋友一定还是朋友。

每对朋友关系用两个整数 $(X_i,Y_i)$ 组成，表示朋友的编号分别为 $X_i,Y_i$。男人的编号是正数，女人的编号是负数。小明的编号是 $1$，小红的编号是 $-1$。

大家都知道，小明和小红是朋友，那么，请你写一个程序求出两公司之间，通过小明和小红认识的人最多一共能配成多少对情侣（包括他们自己）。

输入格式

输入的第一行，包含 $4$ 个空格隔开的正整数 $N,M,P,Q$。

之后 $P$ 行，每行两个正整数 $X_i,Y_i$。

之后 $Q$ 行，每行两个负整数 $X_i,Y_i$。

输出格式

输出一行一个正整数，表示通过小明和小红认识的人最多一共能配成多少对情侣（包括他们自己）。

输入输出样例 #1

输入 #1

4 3 4 2
1 1
1 2
2 3
1 3
-1 -2
-3 -3

输出 #1

2

说明/提示

对于 $30\%$ 的数据，$N,M\le 100$，$P,Q\le 200$；

对于 $80\%$ 的数据，$N,M\le 4\times 10^3$，$P,Q\le 10^4$；

对于 $100\%$ 的数据，$N,M\le 10^4$，$P,Q\le 2\times 10^4$。

C++实现

#include<bits/stdc++.h>
#define r(i,a,b) for(int i=a;i<=b;i++)
using namespace std;int n,m,p,q,u,v,tat,tot;
map<int,int>f;//STl大法好
int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
void judge(int x,int y){f[find(x)]=find(y);}
bool too(int x,int y){return find(x)==find(y);}
int main()
{
	scanf("%d%d%d%d",&n,&m,&p,&q);//输入
	r(i,-1*m,n) f[i]=i;//初始化
	r(i,1,p+q)//直接一起做
	 {
	 	scanf("%d%d",&u,&v);
	 	judge(u,v);
	 }
	r(i,-1*m,-1)
	 if(too(f[i],-1)) tat++;//找连接在一起的
	r(i,1,n)
	 if(too(f[i],1)) tot++;//同理
	printf("%d",min(tat,tot));//几对朋友就是最小值
}

后续

接下来我会不断用C++来实现信奥比赛中的算法题、GESP考级编程题实现、白名单赛事考题实现，记录日常的编程生活、比赛心得，感兴趣的请关注，我后续将继续分享相关内容