版权声明:本文为博主原创文章,未经博主许可允许转载。 https://blog.csdn.net/qq_29600137/article/details/89417997
题目描述
在张老师的学生中,每一名同学都有一个能力值。现在张老师想挑选一个两人小组,他有个奇怪的要求,
需要这两名同学的能力值加起来和张老师一个期望的值一样,现在请你帮忙计算他有多少种挑选学生组成小组的方式,注意不同学生能力值可能相同。
输入描述:
第一行一个整数n(2<=n<=100000)和一个整数t(1<=t<=100000), 表示张老师学生的个数和他期望的值, 接下来一行有n个非负整数,第i个非负整数表示第i名学生的能力值,其中每名学生的能力值都不会超过t。
输出描述:
输出一行,表示张老师能挑选出的小组的方案数
示例1
输入
4 5 2 3 3 2
输出
4
说明
可以选择的方式有(1,2),(1,3),(2,4),(3,4)共4种
示例2
输入
5 4 2 2 2 2 2
输出
10
#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;
vector<int> nums;
int maps[100000+100];
int n, tar;
long long int ans;
int main(void){
cin>> n >> tar;
memset(maps,0,sizeof(maps));
for(int i = 0;i<n;i++){
int tmp;
cin >> tmp;
// nums.push_back(tmp);
if(tmp<=tar){
maps[tmp]++;
}else{
continue;
}
if(maps[tmp]<2){
nums.push_back(tmp);
}
}
// sort(nums.begin(), nums.end());
ans = 0;
for(int i=0;i<nums.size();i++){
if(nums[i]==(tar-nums[i])){
ans += (maps[nums[i]]*(maps[tar-nums[i]]-1));
}else{
ans += (maps[nums[i]]*maps[tar-nums[i]]);
}
}
cout << ans/2 << endl;
return 0;
}