小米OJ 15（和为零的三元组）

最新推荐文章于 2023-12-09 15:31:38 发布

优化少年

最新推荐文章于 2023-12-09 15:31:38 发布

阅读量278

点赞数

分类专栏：小米OJ（“有挑战”难度题集）文章标签： ACM OJ 大学生程序设计大赛小米

本文链接：https://blog.youkuaiyun.com/qq_29117927/article/details/88203466

版权

小米OJ（“有挑战”难度题集）专栏收录该内容

9 篇文章

订阅专栏

本篇博客介绍了一种算法，用于在一个整数数组中寻找所有满足三数之和为零的三元组，并详细展示了实现这一功能的C++代码。通过排序和双指针技巧，该算法能有效避免重复计算，最终输出满足条件的三元组数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

                                            和为零的三元组
                           序号：#15  难度：有挑战  时间限制：1000ms  内存限制：10M

描述
给出一个整数数组, 数组中是否存在任意 3 个数 a, b, c 满足 a + b + c = 0? 找出数组中所有满足以上条件的三元组，最后输出这些三元组的个数（包含相同元素的三元组只计算一次）。

输入
一个包含多个整数（正或负）的字符串，每个整数之间用逗号（,）分隔，如：-1,0,1,2,-1,-4。

输出
输入满足加和结果正好等于 0 的三元组的个数，如对于 -1,0,1,2,-1,-4 有 [-1, 0, 1] 和 [-1, -1, 2]，所以输出 2

输入样例
-1,0,1,2,-1,-4

输出样例
2

#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
	long long int x,a[200005],i,j,s=1,cnt=0,z;
	char t;
	cin>>a[0];
	while(cin>>t)
	{
		cin>>a[s];
		s++;
	}
	sort(a,a+s);
	for(i=s-1;i>=2;i--)
	{
		if((i==(s-1))||(a[i+1]!=a[i]))
		{
			if(a[i]>=0)
			{
				for(j=i-1;j>=1;j--)
				{
					if((j==(i-1))||(a[j+1]!=a[j]))
					{
						if((a[i]+a[j])>=0)
						{
							for(z=j-1;z>=0;z--)
							{
								if((z==(j-1))||(a[z+1]!=a[z]))
								{
									if((a[i]+a[j]+a[z])<0)
									{
										break;
									}
									else
									{
										if((a[i]+a[j]+a[z])==0)
										{
											cnt++;
										}
									}
								}	
							}
						}
						else
						{
							break;
						}
					}
				}
			}
			else
			{
				break;
			}
		}
	}
	cout<<cnt<<endl;
}