美团8.13笔试

t4 合法的三元组

// a[i] = 3*a[j] - a[k] && i<j<k

int main()
{
	int n, count=0;
	cin >> n;
	for (int i=0; i<n; i++) cin >> a[i];

	unordered_map<int, int> mp;
	
	mp[a[0]] = 1;

	for (int j = 1; j<n-1; j++) {
		for (int k=j+1; k<n; k++) {
			count += mp[3*a[j]-a[k]];
		}
		if (j == n-2) break;

		mp[a[j]] += 1;
	}
	cout << count;
	return 0;
}


全部评论
请问你这题是ac的代码吗?
点赞
送花
回复
分享
发布于 2022-08-24 16:25 四川

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务