【“纽劢科技杯”第十六届同济大学程序设计竞赛暨上海邀请赛同步赛】B-张老师组队

 

题目描述

    在张老师的学生中，每一名同学都有一个能力值。现在张老师想挑选一个两人小组，他有个奇怪的要求，
    需要这两名同学的能力值加起来和张老师一个期望的值一样，现在请你帮忙计算他有多少种挑选学生组成小组的方式，注意不同学生能力值可能相同。

输入描述:

第一行一个整数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;
}