题意: 给你一个图,如果ij 为+ 表明 两点之间有一条有向边,那么你要找出所有的三元环的个数。
思路: 先吐槽一下,这个题要文件读入读出,坑死。。。
那么如果暴力的方法其实是n3 for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) for(int k=1;k<=n;k++) if(mp[i][j]&&mp[j][k]&&mp[k][i]) ans++;
但是n为1500,那么暴力就不存在了,但是我们可用bitset 优化掉 一维,取& 就可以得到结果了。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N =1505;
bitset< N > in[N];
bitset< N >out[N];
bitset< N > tmp;
char mp[N][N];
int n;
int main()
{
freopen("triatrip.in","r",stdin);
freopen("triatrip.out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%s",mp[i]+1);
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
if(mp[i][j]=='+'){
out[i][j]=1;
in[j][i]=1;
}
}
}
ll ans=0;
for(int u=1;u<=n;u++){
for(int v=1;v<=n;v++){
if(u==v) continue;
if(mp[u][v]=='+'){
//cout<<"u "<<u<<" v "<<v<<endl;
tmp.set();
tmp.reset(u); tmp.reset(v);
tmp=(tmp&in[u]&out[v]);
ans+=tmp.count();
//cout<<"tmpcnt "<<tmp.count()<<endl;
}
}
}
ans/=3;
cout<<ans<<endl;
return 0;
}
/*
4
--+-
+--+
-+--
--+-
4
+++-
+++-
+++-
---+
4
++++
++++
++++
++++
3
+++
+++
+++
*/