考研机试 三元组

时间:2024-01-27 22:35:17

考研机试 三元组

给定一个长度为 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);
    }
}

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