Counting Cliques
题意:给你n个点,m条边,求大小为s(s<=10)的团有多少个,每个点的度最多20。看到数据这么小,想着各种bitset暴力,想着把3元团预处理,再预处理4元团等等。写了两个多小时发现复杂度算错了,9元团的个数能达到1e6,这样我们两层循环预处理就不行了。
看题解原来是ssssb的爆搜,题目给你degree<=20就是在暗示你可以爆搜,于是爆搜写了一发结果TLE,What f*ck。再加了个小小的优化然后AC。。
TLE代码:
int n,m,s,sum;
int vis[101],g[101][101];
int tot,head[101];
struct egde
{
int to,next;
} e[N];
void add(int u,int v)
{
e[tot].to=v,e[tot].next=head[u];
head[u]=tot++;
}
void solve(int u,int num)
{
if(num==s)
{
sum++;
return ;
}
for(int i=head[u];i+1;i=e[i].next)
{
int v=e[i].to,flag=1;
if(vis[v]) continue;
for(int i=1;i<=n;i++) if(vis[i]&&!g[v][i]) flag=0;
if(flag)
{
vis[v]=1;
solve(v,num+1);
vis[v]=0;
}
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&s);
memset(vis,0,sizeof(vis));
memset(g,0,sizeof(g));
int u,v;
tot=sum=0;
memset(head,-1,sizeof(head));
for(int i=1; i<=m; i++)//建单向边减少搜索次数
{
scanf("%d%d",&u,&v);
if(u>v) swap(u,v);
add(u,v);
g[v][u]=g[u][v]=1;
}
for(int i=1; i<=n; i++)
if(!vis[i])
{
vis[i]=1;
solve(i,1);
// vis[i]=0;
}
printf("%d\n",sum);
}
return 0;
}
上面的solve里面每次都要验证100个点,这样复杂度是100*10*100*100*T。
把这个地方优化一下, 把已标记过的放入栈里,每次判10个就可以了。
int n,m,s,sum;
int vis[101],g[101][101],Stack[101];
int tot,head[101];
struct egde
{
int to,next;
} e[N];
void add(int u,int v)
{
e[tot].to=v,e[tot].next=head[u];
head[u]=tot++;
}
void solve(int u,int num)
{
if(num==s)
{
sum++;
return ;
}
for(int i=head[u];i+1;i=e[i].next)
{
int v=e[i].to,flag=1;
for(int j=1;j<=num&&flag;j++) if(!g[v][Stack[j]]) flag=0;
if(flag)
{
Stack[num+1]=v;
solve(v,num+1);
}
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&s);
memset(g,0,sizeof(g));
int u,v;
tot=sum=0;
memset(head,-1,sizeof(head));
for(int i=1; i<=m; i++)//建单向边减少搜索次数
{
scanf("%d%d",&u,&v);
add(u,v);
g[v][u]=g[u][v]=1;
}
for(int i=1; i<=n; i++)
{
Stack[1]=i;
solve(i,1);
}
printf("%d\n",sum);
}
return 0;
}