BUPT复试专题—三元组(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)，表示数组长度，然后输入这个数组。

输出

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

样例输入

2
2
0 0
5
1 1 1 2 1

样例输出

8
16

来源

2016机考A题

#include<iostream>
#define maxn 105
using namespace std;
 
int a[maxn];
 
int main()
{
    int tes,n;
    int i,j,k;
    while(cin>>tes)
    {
        while(tes--)
        {
            cin>>n;
            for(i=0; i<n; i++)
                cin>>a[i];
 
            int cnt=0;
            for(i=0; i<n; i++)  
                for(j=0; j<n; j++)
                    for(k=0; k<n; k++)
                    {
                        if(a[i]+a[j]==a[k])
                            cnt++;
                    }
            cout<<cnt<<endl;
        }
    }
    return 0;
}