【JZOJ4857】Tourist Attractions（Bitset）

题意：给定一个n个点的无向图，求这个图中有多少条长度为4的简单路径。

n<=1500

思路：

#include<map>
#include<set>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<bitset>
#define MAXN 1510
#define MAXM 4100000
using namespace std;

typedef long long ll;
bitset<MAXN> a[MAXN],tmp;

int head[MAXM],vet[MAXM],next[MAXM],n,i,j,tot;
ll size[MAXN];
ll ans;

void add(int a,int b)
{
  next[++tot]=head[a];
  vet[tot]=b;
  head[a]=tot;
}

int main()
{
    //freopen("jzoj4857.in","r",stdin);
    //freopen("jzoj4857.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("
");
        for(int j=1;j<=n;j++) 
        {
            char ch;
            scanf("%c",&ch);
            if(ch=='1')
             {
                add(i,j);
                size[i]++;
                a[i][j]=1;
             }
                
        }
    }
    for(int i=1;i<=n;i++)
     for(int e=head[i];e;e=next[e])
     {
         int v=vet[e];
         ans+=(size[i]-1)*(size[v]-1);
         tmp=a[i]&a[v];
         ans-=tmp.count();
     }
     printf("%lld
",ans);
     return 0;
}