HDU 6016 Count the Sheep （思路）

最新推荐文章于 2018-09-13 11:44:39 发布

「已注销」

最新推荐文章于 2018-09-13 11:44:39 发布

阅读量242

点赞数

分类专栏：思路

本文链接：https://blog.csdn.net/wing_wuchen/article/details/75104966

版权

思路专栏收录该内容

15 篇文章 0 订阅

订阅专栏

思路：

题目要求找 a-b-c-d
有因为题目给出的是一个二分图，所以这个四元关系类似这样：0 1 0 1 或 1 0 1 0，
所以可知a和d一定不同
所以我们只要枚举中间的边，然后把（b的朋友数-1）*（c的朋友数-1）加到答案中。
（最后注意乘二，因为颠倒顺序又能出来一组）
（本题貌似没有重边很好要不更麻烦了。。）

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string.h>
#include <cmath>
#include <queue>
#define eps 1e-8
typedef long long int lli;
using namespace std;

struct edge{
    int to,next;
}ed[201000];int cnte;
int head[101000];
void ae(int x,int y){
    ed[++cnte].to = y;
    ed[cnte].next = head[x];
    head[x] = cnte;
}
int m,n,k,a,b;
int cnt[210000];

int main(){
    int t;
    cin>>t;
    while(t--){
        memset(head,-1,sizeof(head));
        memset(cnt,0,sizeof(cnt));
        cnte = 0;
        int n,m,a,b,k;
        scanf("%d%d%d",&n,&m,&k);
        for(int i = 1;i <= k;i++){
            scanf("%d%d",&a,&b);
            ae(a,b+n);cnt[a]++;cnt[b+n]++;
        }
        lli ans = 0;
        for(int i = 1;i <= n;i++){
            for(int j = head[i];j!=-1;j=ed[j].next){
                int to = ed[j].to;
                ans += (cnt[i]-1)*(cnt[to]-1);
            }
        }
        printf("%lld\n",ans*2);
    }
}