题目描述
人与人之间的关系错综复杂,常常会出现一个叫作共同好友的东西。所以,贴心的 QQ 就提供了这样一个功能,可以显示你与某人(不一定是好友)有多少个共同好友。但是,当用户量逐渐增大,好友关系网不断复杂化,共同好友计算的效率就变得十分重要了。
你刚刚和腾讯公司签约,获得了共同好友计算的开发资格。
输入格式
第一行有两个整数 n,m (1≤n≤40 000,1≤m≤40 000)。分别表示用户数量和好友关系数量。方便起见,用户编号为 1 到 n。
接下来 m 行,每行两个整数用空格隔开 u,v (1≤u,v≤n,u≠v),表示 u 和 v 是好友。数据保证不会出现两对相同的 u,v。
接下来一行一个整数 q (1≤q≤40 000) 表示查询数。
接下来 q 行,每行两个整数 s,t (1≤s,t≤n,s≠t),表示询问的对象。
小数据规模说明:
- 对于前 20% 的数据有 1≤n≤10,1≤q≤200。
- 对于前 40% 的数据有 1≤n≤1 000。
输出格式
对于每组询问,输出这两个人有多少个共同好友。
样例
input
3 3 1 2 1 3 3 2 2 1 3 3 2output
1 1input
4 4 1 2 1 4 2 3 3 4 3 1 3 1 4 2 4output
2 0 2input
3 2 1 2 1 3 2 2 3 2 3output
1 1
思路:用邻接矩阵存储,为了加快运算,第一个想到的就是位运算,所以采用bitset容器实现。一开始用vector里套bitset导致内存超了,后来直接用二维bitset,成功解答,代码如下:
#include <iostream>
#include <bitset>
#include <vector>
using namespace std;
int main()
{
int n, m, q, u, v;
bitset<40000> C[40000];
cin>>n>>m;
for(int i=0; i<m; ++i){
cin>>u>>v;
C[u-1].set(v-1);
C[v-1].set(u-1);
}
cin>>q;
for(int i=0; i<q; ++i){
cin>>u>>v;
cout<<(C[u-1]&C[v-1]).count()<<endl;
}
return 0;
}