Codeforces Gym 100342J Problem J. Triatrip 求三元环的数量 bitset
Problem J. Triatrip
Time Limit: 20 Sec
Memory Limit: 256 MB
题目连接
http://codeforces.com/gym/100342/attachments
Description
The travel agency “Four Russians” is offering the new service for their clients. Unlike other agencies that only suggest one-way or roundtrip for airline tickets to their customers, “Four Russians” offers the brand new idea — triatrip. Triatrip traveler starts in some city A, flies to some city B, then flies to some city C, and returns to the city A.
Now the managers of the agency started to wonder, how many different triatrips they can offer to their customers. Given a map of all possible flights, help them to find that out.
Input
The first line of the input file contains two integer numbers n — the number of cities that are served by airlines that agree to sell their tickets via the agency (3 ≤ n ≤ 1500). The following n lines contain a sequence of n characters each — the j-th character of the i-th line is ‘+’ if it is possible to fly from the i-th city to the j-th one, and ‘-’ if it is not. The i-th character of the i-th line is ‘-’.
Output
Output one integer number — the number of triatrips that the agency can offer to its customers.
Sample Input
4
--+-
+--+
-+--
--+-
Sample Output
2
HINT
题意
给出邻接矩阵,有向图,找出三元环的个数
题解:
bitset 暴力枚举一条边,出度入度的集合用到bitset
代码:
1 //作者:1085422276 2 #include <cstdio> 3 #include <cmath> 4 #include <cstring> 5 #include <ctime> 6 #include <iostream> 7 #include <algorithm> 8 #include <set> 9 #include <vector> 10 #include <sstream> 11 #include <queue> 12 #include <typeinfo> 13 #include <fstream> 14 #include<bits/stdc++.h> 15 #include <map> 16 #include <stack> 17 typedef long long ll; 18 using namespace std; 19 const int inf = 10000000; 20 inline ll read() 21 { 22 ll x=0,f=1; 23 char ch=getchar(); 24 while(ch<'0'||ch>'9') 25 { 26 if(ch=='-')f=-1; 27 ch=getchar(); 28 } 29 while(ch>='0'&&ch<='9') 30 { 31 x=x*10+ch-'0'; 32 ch=getchar(); 33 } 34 return x*f; 35 } 36 ll exgcd(ll a,ll b,ll &x,ll &y) 37 { 38 ll temp,p; 39 if(b==0){ 40 x=1;y=0; 41 return a; 42 } 43 p=exgcd(b,a%b,x,y); 44 temp=x;x=y;y=temp-(a/b)*y; 45 return p; 46 } 47 //******************************* 48 #define N 1510 49 bitset<N> in[N],out[N],he; 50 char mp[1501][1501]; 51 int main() 52 { 53 54 freopen("triatrip.in","r",stdin); 55 freopen("triatrip.out","w",stdout); 56 int n; 57 n=read(); 58 for(int i=1;i<=n;i++) 59 scanf("%s",mp[i]); 60 for(int i=1;i<=n;i++) 61 { 62 for(int j=1;j<=n;j++) 63 { 64 if(i==j)continue; 65 if(mp[i][j-1]=='+') 66 { 67 in[i][j]=true; 68 out[j][i]=true; 69 } 70 } 71 } 72 ll ans=0; 73 for(int i=1;i<=n;i++) 74 { 75 for(int j=1;j<=n;j++) 76 { 77 if(i==j)continue; 78 if(mp[i][j-1]=='+') 79 he=(in[j])&(out[i]),ans+=he.count(); 80 } 81 } 82 cout<<ans/3<<endl; 83 return 0; 84 }