小b有一个仅包含非负整数的数组a,她想知道有多少个三元组(i,j,k),满足i<j<k且a[i],a[j],a[k]可能作为某个三角形的三条边的边长。
输入:第一行输入一个正整数n,表示数组a中元素个数;
第二行n个非负整数,表示a中元素,以空格隔开;
其中0<n≤1000,a中任意元素a[i]满足0≤a[i]≤1000。
输出:输出一个数,表示满足题意的三元组个数
样例输入 Copy
4
2 2 3 4
样例输出 Copy
3
法一:会超时,采用三重for循环,最容易想到,但运行时间过长,会超出时间限制。
#include <iostream>
using namespace std;
int main()
{
int n;
cin>>n;
int a[n];
if(n<3) return 0;
for(int i=0;i<n;i++)
cin>>a[i];
int count=0;
for(int i=0;i<n-2;i++)
{
for(int j=i+1;j<n-1;j++)
{
for(int k=j+1;k<n;k++)
{
if(a[i]>0&&a[j]>0&&a[k]>0&&a[i]+a[j]>a[k]&&a[i]+a[k]>a[j]&&a[j]+a[k]>a[i])
count++;
}
}
}
cout<<count<<endl;
return 0;
}
法二:会超时,虽然看似将法一的三重for循环降为一重,但本质上比较次数并没有得到改善,所以依旧会超时。
#include <iostream>
using namespace std;
int main()
{
int n;
cin>>n;
int a[n];
if(n<3) return 0;
for(int i=0;i<n;i++)
cin>>a[i];
int count=0;
for(int i=0;i<n-2;i++)
{
for(int j=i+1;j<n-1;j++)
{
for(int k=j+1;k<n;k++)
{
if(a[i]>0&&a[j]>0&&a[k]>0&&a[i]+a[j]>a[k]&&a[i]+a[k]>a[j]&&a[j]+a[k]>a[i])
count++;
}
}
}
cout<<count<<endl;
return 0;
}
法三:正确,思路是先将数组元素进行排序,再判断是否能组成三角形,这样可以避免一些多余的比较,很大程度降低时间复杂度,其中排序采用的是归并排序,最好、最坏、平均时间复杂度均为0(nlogn),是目前为止比较好的排序算法,从而不会超时。
#include <iostream>
using namespace std;
void merge_array(int array[],int first,int mid,int last,int temp[])//归并排序中的归并,即将一个数组排好序的两部分合并到一起
{
int i=first,j=mid+1,k=0;
while(i<=mid&&j<=last)
{
if(array[i]<=array[j])
{
temp[k++]=array[i++];
}
else
{
temp[k++]=array[j++];
}
}
while(i<=mid)
{
temp[k++]=array[i++];
}
while(j<=last)
{
temp[k++]=array[j++];
}
for(int i=0;i<k;i++)
array[first+i]=temp[i];
}
void merge_sort(int data[],int first,int last,int temp[])//归并排序,对数组元素进行排序
{
if(first<last)
{
int mid=(first+last)/2;
merge_sort(data,first,mid,temp);
merge_sort(data,mid+1,last,temp);
merge_array(data,first,mid,last,temp);
}
}
int main()
{
int n,count=0;
cin>>n;
if(n<3)return 0;
int a[n],temp[n];
for(int i=0;i<n;i++)
cin>>a[i];
merge_sort(a,0,n-1,temp);
for(int i=0;i<n-2;i++)
{
for(int j=i+1;j<n-1;j++)
{
for(int k=j+1;k<n;k++)
{
if(a[i]>0&&a[i]+a[j]>a[k])
{
count++;
}
else
break;
}
}
}
cout<<count<<endl;
return 0;
}