考研机试 三元组

考研机试 三元组

给定一个长度为 m 的数组 a0,a1,...,am−1。

如果数组中有 ai+aj=ak其中 i,j,k 大于等于 0 并且小于 m，则称 (ai,aj,ak)为一个三元组。

现在，给定你数组 a，请你计算其中三元组的个数。

例如，当 m=2，数组 a 为 {0,0}时，所有三元组为：

(a0,a0,a0)
(a0,a0,a1)
(a0,a1,a0)
(a0,a1,a1)
(a1,a0,a0)
(a1,a0,a1)
(a1,a1,a0)
(a1,a1,a1)

共计 8个三元组。

输入格式

第一行包含一个整数 n，表示共有 n组测试数据。

每组数据第一行包含整数 m，表示数组长度。

第二行包含 m个整数，表示数组。

输出格式

每组数据输出一行一个答案，表示三元组个数。

数据范围

1≤n≤10,

1≤m≤50，

数组元素取值范围 [0,100]。

#include <iostream>
#include <vector>
using namespace std;
int main(){
    int n,m;
    cin>>n;
    for(int i=0;i<n;++i){
       cin>>m;
        vector<int> vec1(m);
        vector<int>::iterator it;
        for(int j=0;j<m;++j){//数据输入完成
            cin>>vec1[j];
        }
        int sum=0;
        for(int k=0;k<m;++k){
            for(int q=0;q<m;++q){
                int temp=vec1[k]+vec1[q];
                for(it=vec1.begin();it!=vec1.end();++it){
                    if(*it==temp){
                        sum++;
                    }
                }
            }
        }
        printf("%d\n",sum);
    }
}

思路比较简单使用暴力循环+动态数组就能解决 甚至普通数组三重循环也能解决