标签:bitset
题解:我们设f[i][j]表示以点i为起点,j为路径长度的简单路径数量。
显然有f[i][1]=1,
f[i][2]=∑所有j与i相连f[j][1]
,
f[i][3]=∑所有j与i相连f[j][2]−1
。
注意到在一条长度为4的路径中,第一个点不能与第三,第四个点相等,第二个点不能与第四个点相等。
所以
f[i][4]=∑所有j与i相连f[j][3]−j的度数+1−包含i,j的三元环个数
。
上述做法时间复杂度为O(n^3),只能拿70分。
我们对其进行优化,发现一个三元环会在计算f[i][4]时被计算6次(自己脑补),于是我们可以改为
f[i][4]=∑所有j与i相连f[j][3]−j的度数+1
,最后答案减去6*三元环个数(虽然这并没有什么用)。
对于三元环个数的计算,我们使用bitset,这样时间复杂度可降到O(n^3/64),就可以过了。
//created by Talex Yeung,all rights reserved.
#include<cstdio>
#include<ctime>
#include<algorithm>
#include<cstring>
#include<bitset>
#define maxn 1550
using namespace std;
typedef long long ll;
char s[maxn];
int n,l[maxn];
ll f[maxn][5],ans=0ll,zzk=0ll,cnt=0ll,hz=0ll;
bitset<maxn> b[maxn],tmp;
int main()
{
freopen("1.txt","r",stdin);
scanf("%d",&n),memset(f,0,sizeof(f));
for (int i=1;i<=n;i++)
{
scanf("%s",s);
for (int j=1;j<=n;j++) if ((b[i][j]=s[j-1]-'0')==1) l[i]++;
}
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
if (b[i][j]) tmp=b[i]&b[j],hz+=tmp.count();
for (int i=1;i<=n;i++) f[i][1]=1;
for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) if (b[i][j]) f[i][2]+=f[j][1];
for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) if (b[i][j]) f[i][3]+=f[j][2]-1;
for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) if (b[i][j]) f[i][4]+=f[j][3]-l[j]+1;
for (int i=1;i<=n;i++) ans+=f[i][4];
printf("%lld",ans-hz);
}