BestCoder Round #92 1002 Count the Sheep —— 枚举+技巧

题目链接：http://bestcoder.hdu.edu.cn/contests/contest_showproblem.php?cid=748&pid=1002

题解：

做题的时候只是想到用dfs暴搜，结果超时了。（刚学dfs时以为它无所不能，后来渐渐不太喜欢了，因为太暴力了，经常超时）

TLE wa代码如下：

#include<bits/stdc++.h>

using namespace std;

vector<int>v0[100005],v1[100005];
int vis[2][100005],ans;

void dfs(int k, int flag, int step)//flag标记男女，0为男，1为女。
{
if(step==4) {ans++; return ;}

if(flag)
{
int sum = v1[k].size();
for(int i = 0; i<sum; i++)
if(!vis[0][v1[k][i]])
{
vis[0][v1[k][i]] = 1;
dfs(v1[k][i],0,step+1);
vis[0][v1[k][i]] = 0;
}
}

else
{
int sum = v0[k].size();
for(int i = 0; i<sum; i++)
if(!vis[1][v0[k][i]])
{
vis[1][v0[k][i]] = 1;
dfs(v0[k][i],1,step+1);
vis[1][v0[k][i]] = 0;
}
}
}

int main()
{
int n,m,k,T,t0,t1;
scanf("%d",&T);
while(T--)
{
ans = 0;
scanf("%d%d%d",&n,&m,&k);
for(int i = 1; i<=n; i++) v0[i].clear();
for(int i = 1; i<=m; i++) v1[i].clear();
memset(vis,0,sizeof(vis));
for(int i = 0; i<k; i++)
{
scanf("%d%d",&t0,&t1);
v0[t0].push_back(t1);
v1[t1].push_back(t0);
}

for(int i = 1; i<=n; i++)
{vis[0][i] = 1; dfs(i,0,1); vis[0][i] = 0;}
for(int i = 1; i<=m; i++)
{vis[1][i] = 1; dfs(i,1,1); vis[1][i] = 0;}

printf("%d\n",ans);
}
return 0;
}

后来大致看了题解，就觉得可以逐个点枚举，从第一个到第四个，用for循环，但仔细想想，这跟dfs没用本质的区别，只不过递推与递归的实现不同。

于是又认真看了题解。发现他不是从第一个点开始枚举的，而是从第二第三个点枚举的，为什么呢？其实画一下图就可以很好理解。四个点构成三条边，如果从第一个点开始枚举，那么所有人都必须知道朋友是谁，才能继续枚举下去，用图的话来说，是只有知道下一个点是谁，才能构成线段，所以只能一个一个枚举。但是如果从第二第三个开始枚举，即从中间的那条边开始，即枚举中间那条边，那么只需要一个性别知道朋友是谁和个数，另一个性别知道朋友的个数，则直接可以分别用（个数-1），即减去当前已确定的对方，就是剩下的对方，然后相乘。

上述解释可能很模糊，但可以抽象成一种思维方法。就好像第二个点和第三个点处在中间，其两边都还连有点，那么这两个点的地位相对第一第四个点来说要刚，因为其起到的作用，或影响范围更广。所以就挑选影响范围广的进行操作，那么气效率也更高。

学习之处：大致来讲，挑选影响范围广的进行操作，那么其效率会更高。

代码如下：

#include<bits/stdc++.h>

using namespace std;

vector<int>v[100005];//v记录男生的朋友
int cnt[100005];//cnt记录女生的朋友个数
int main()
{
int T, n,m,k,x,y;
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&n,&m,&k);
memset(cnt,0,sizeof(cnt));
for(int i = 1; i<=n; i++)
v[i].clear();

for(int i = 0; i<k; i++)
{
scanf("%d%d",&x,&y);
v[x].push_back(y);
cnt[y]++;
}

long long ans = 0;//用int会溢出

for(int i = 1; i<=n; i++)//枚举男生，即第三个点
{
int sum = v[i].size();//这个男生的朋友个数
for(int j = 0; j<sum; j++)//枚举与之为朋友的女生，即第二个点
{
/*这条公式非常精彩。其实第一第二个循环实际是枚举中间那条边。
首先中间的边已经确定，那么只需要知道第二第三个点的朋友个数，
然后再减去对方，就是剩下的朋友数，再相乘，就是在中间的边已确立
的情况下，不同连线方式的个数。*/
ans += (cnt[v[i][j]]-1) * (sum-1);
}
}

printf("%lld\n",ans*2);
}
return 0;
}