2016年北邮机试真题——三元组

问题描述：

给你一个长度为m的数组（数组元素从0到m-1），如果数组里有a[i]+a[j]==a[k](i,j,k大于等于0并且小于m)，便称之为三元组。现在给你一个数组，让你求三元组的个数。

例如m为2，里面的元素为（0,0）

那么三元组为

（a[0],a[0],a[0]）

（a[0],a[0],a[1]）

（a[0],a[1],a[0]）

（a[0],a[1],a[1]）

（a[1],a[0],a[0]）

（a[1],a[0],a[1]）

（a[1],a[1],a[0]）

（a[1],a[1],a[1]）

输出答案为8.

输入：

输入正整数N，表示N例测试。接着输入N组数据，每组输入m(1<=m<=50)，表示数组长度，然后输入这个数组。

输出：

对每组输入数据，输出三元组的个数。

Sample Input

2

2

0 0

5

1 1 1 2 1

Sample Output

8

16
 

算法实现：

#include<iostream> using namespace std; int main() {     int m;     cin>>m;     while(m>0)     {         int n;         int a[100];         cin>>n;         for(int i=0;i<n;i++)         {             cin>>a[i];                          }         int count=0;         for(int i=0;i<n;i++)         {             for(int j=0;j<n;j++)             {                 for(int k=0;k<n;k++)                 {                     if(a[i]+a[j]==a[k])                     {                         count++;                         }                     }                              }             }         cout<<count<<endl;             m--;     }         return 0; }