HDU 6016 - Count the Sheep

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 #define LL long long
 4 const int MAXN = 100005;
 5 int t, n, m, k;
 6 int C[MAXN];//女羊的朋友
 7 vector<int> G[MAXN]; //男羊->女羊
 8 int main()
 9 {
10     scanf("%d", &t);
11     while (t--)
12     {
13         memset(C, 0, sizeof(C));
14         scanf("%d%d%d", &n, &m, &k);
15         for (int i = 1; i <= n; i++) G[i].clear();
16         memset(C, 0, sizeof(C));
17         for (int i = 1; i <= k; i++)
18         {
19             int x, y; scanf("%d%d", &x, &y);
20             C[y]++;
21             G[x].push_back(y);
22         }
23         LL ans = 0;
24         for (int x = 1; x <= n; x++)
25         {
26             LL A = G[x].size()-1;
27             for (int j = 0; j < G[x].size(); j++)
28                 ans += A*(C[G[x][j]]-1);
29         }
30         printf("%lld\n", 2*ans);
31     }
32 }