HDU-6184-Counting Stars(广西邀请赛C题)(数据结构优化)

题目链接

题意:给你一张图，然后问你有多少个A-数据结构具体结构就是一个正方形里面加一条线。

思路:就是求出所有的三元环，然后组合一下。具体想法就是，对于每一条边，然后求出有多少个点能和这条边组成三角形。然后组合一下就好了。

具体操作:直接枚举边，然后再去枚举每个端点，我先这么枚举了一下，发现不行，T了。然后我们就可以换一种角度，去枚举每一个点，然后在去枚举每一条河这个点有关的边，如果这个点被枚举过了，就可以不用枚举了，这样省下了很多时间。另外这里还用了一个小操作，很骚。就是对于判断一条边是不是存在，可以直接手动把这条边给hash了。具体看代码。

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>

#include <vector>
#include <map>
#include <queue>
#include <stack>
#include <set>

using namespace std;

#define pb push_back
#define pi acos(-1)

const long long mod = 1000000007;

#define maxn 111111
#define maxm 11111
vector<int> gg[maxn];
int vis[maxn];
int nex[maxn];
int out[maxn];
int n, m;
set<long long> ss;
int main()
{

long long ans, tot;
int limt;
while(scanf("%d%d", &n, &m) != EOF) {
int limt = sqrt(m);
ss.clear();
for(int i = 1; i <= n; i++) {
gg[i].clear();
vis[i] = out[i] = nex[i] = 0; //清空所有
}
int x, y;
for(int i = 1; i <= m; i++) {
scanf("%d%d" , &x, &y);
gg[x].pb(y); //放入边
gg[y].pb(x);
out[x]++;
out[y]++;
ss.insert((long long)x * n + y);  //hash边
ss.insert((long long)y * n + x);
}
ans  = 0;
int xx, yy, zz;
for(int i = 1; i <= n; i++) {
xx = i;
vis[xx] = 1;
for(int j = 0; j < gg[xx].size(); j++)
nex[gg[xx][j]] = xx;   //nex数组是一个可持续化的数组，可以用来储存当前点所连点边
for(int j = 0; j < gg[xx].size(); j++) {
tot = 0;
yy = gg[xx][j];
if(vis[yy]) continue;
if(out[yy] <= limt) { // 如果yy点所连点边超过log(m)就暴力枚举，如果超过就用set优化
for(int k = 0; k < gg[yy].size(); k++) {
zz = gg[yy][k];
if(nex[zz] == xx) tot++;
}
}
else {
for(int k = 0; k < gg[xx].size(); k++) {
zz = gg[xx][k];
if(ss.find((long long) zz * n + yy) != ss.end()) tot++;   //set优化
}
}
ans += tot * (tot - 1) / 2;
}

}
printf("%lld\n", ans);

}

return 0;
}