B - Linear Algebra Test Gym - 101502B

https://vjudge.net/contest/285982#problem/H

题意：

给出n个矩阵的行数与列数，问有多少对矩阵可以相乘。
且矩阵可以相乘的条件是，前一个矩阵的列数=后一个的行数。

注意：

数据是1e9所以正常做法会超时，本题我用map求解

#include<bits/stdc++.h>
using namespace std;
int a[120000][2];
int main()
{
  int t,i,n;
  map < int, int> m;
  scanf("%d",&t);
  while(t--)
  {
     int cc=0;
     m.clear();
    scanf("%d",&n);
    for(i=0;i<n;i++)
    {

      for(int j=0;j<2;j++)
      {
        scanf("%d", &a[i][j]);
      }

    }
    for(i=0;i<n;i++)
    {
     m[a[i][0]]++;
    }
    long long ans = 0;
    for(i=0;i<n;i++)
    {
      ans += m[a[i][1]];

    }



    printf("%lld\n",ans);

  }
  return 0;

}