思路:N的范围为1000,所以四重for循环枚举显然不行。所以我们可以先将两个不同数字相加的结果算出来,然后问题就变为在这个数组中找到两个数字相加等于0的问题。
如果在枚举一个数字,在这个集合中寻找它的负数,可是我们这样复杂度依旧为n^2*n^2=n^4会超时。我们有两种做法(先对数组排序):
<1> 二分查找,枚举一个数二分查找他的负数,复杂度为,这样就能过了。可是你在仔细想想这四个数字不能重叠,可能会出现A+B+B+C=0的情况,所以这样写是有BUG的,所以我们要对每两个数字的和做标记表示他是由哪两个数字相加得出来得(用个结构体就可以了)。再想一下二分可能得情况,对于当前枚举的数字sum[i],查找数组为sum查找值key=-sum[i],①key<sum[m],使L=m+1向右缩区间 ②key>sum[m],使R=m-1向右缩区间 ③key==sum[m],这种时候我们要判断是不是sum[i]=A+B,sum[m]=B+X,sum[i]+sum[m]=A+B+B+X=0的情况,如果不是就找到了答案直接跳出到最外层输出答案就行了,否则的话跳出二分,去枚举下一个数字就可以(为什么?解释在下面关键点中)。
<2>我们也可以采取尺取法,复杂度为。然后定义两个指针i从前往后扫,j从后往前扫,①sum[i]+sum[j]<0,这时候让i++②sum[i]+sum[j]>0,让j--③sum[i]+sum[j]=0,这时候让i++,j--(为什么?解释在下面关键点中)
关键点:为什么A+B+B+X=0的时候可以直接跳出二分去枚举下一个数字(i++,j--)呢,这样会错过正确答案吗?
对于sum1=A+B,sum2=B+X,首先是没答案的时候我们跳过这一种情况自然没什么问题
如果有答案的话我们这样会不会错过正确答案呢,如果有答案的话肯定有A+B+C+D=0或者其他的情况W+X+Y+Z=0。我们这样可能对W+X+Y+Z自然没影响。
对A+B+C+D=0是有影响的,当sum1=A+B,sum2=B+X我们的写法会跳过这种的情况去枚举下一个数,可能会导致C+D失配,错过sum1=A+B,sum2=C+D这种这种正确答案。可是对于A+B+C+D=0的情况,还有sum1=A+C,sum2=B+D,sum1=A+D,sum2=B+C两种情况能得到正确答案。
可能这个时候有同学会问会不会每次都错过导致答案错误?我认为:不会的,因为这样的数据不存在。这个证明和观点可能会有问题,如果有错误的地方请大佬指出,谢谢。
首先我们假设有唯一的答案A+B+C+D=0,要想使我们这种枚举+二分的方法产生错误,肯定要跳过A,B,C,D任意两个数字组合在一起的情况(如果有一种没被跳过我们就能得到正确答案,如C+D没被跳过,那么一定能二分找到A+B从而得到正确答案,即下图(前两个数相加为我们枚举的数字,后两个数字相加为我们二分找的数字)
此时会有A+B+D+J=0,A+C+D+I=0等等,也必须也避免这些情况的出现才能使我们答案错误,将这些情况列出来以后还会同理衍生出来更多的式子,一直循环无尽,所以不存在这样的数据。
二分:
#include<bits/stdc++.h>
using namespace std;
struct Sum
{
int x,y,s;
}sum[1005*1005];
int n,a[1005];
bool cmp(struct Sum a,struct Sum b)
{
return a.s<b.s;
}
int main()
{
scanf("%d",&n);
for(int i = 0; i < n; i++) scanf("%d",&a[i]);
int cnt=0;
for(int i = 0; i < n; i++)
{
for(int j = i + 1; j < n; j++)
{
sum[cnt].s=a[i]+a[j];
sum[cnt].x=i;sum[cnt].y=j;
cnt++;
}
}
sort(sum,sum+cnt,cmp);
bool flag=false;
for(int i = 0; i < cnt; i++)
{
int l=0,r=cnt-1,m = l + (r - l) / 2,key=-sum[i].s;
while(l <= r)
{
if(sum[m].s < key)
{
l = m + 1;
}
else if(sum[m].s > key)
{
r = m - 1;
}
else if(sum[m].s == key)
{
if( (sum[i].x != sum[m].x) && (sum[i].x != sum[m].y)
&& (sum[i].y != sum[m].x) && (sum[i].y != sum[m].y) )
{
flag = true;
}
break;
}
m = l + (r - l) / 2;
}
if(flag) break;
}
if(flag) printf("Yes\n");
else printf("No\n");
return 0;
}
尺取:
#include<bits/stdc++.h>
using namespace std;
struct Sum
{
int x,y,s;
}sum[1005*1005];
int n,a[1005];
bool cmp(struct Sum a,struct Sum b)
{
return a.s<b.s;
}
int main()
{
scanf("%d",&n);
for(int i = 0; i < n; i++) scanf("%d",&a[i]);
int cnt=0;
for(int i = 0; i < n; i++)
{
for(int j = i + 1; j < n; j++)
{
sum[cnt].s=a[i]+a[j];
sum[cnt].x=i;sum[cnt].y=j;
cnt++;
}
}
sort(sum,sum+cnt,cmp);
bool flag=false;
int l = 0,r = cnt - 1;
while(l < r)
{
if(sum[l].s + sum[r].s < 0) l++;
else if(sum[l].s + sum[r].s > 0) r--;
else if(sum[l].s + sum[r].s == 0)
{
if( (sum[l].x != sum[r].x) && (sum[l].x != sum[r].y)
&& (sum[l].y != sum[r].x) && (sum[l].y != sum[r].y) )
{
flag = true;break;
}
else
{
l++;r--;
}
}
}
if(flag) printf("Yes\n");
else printf("No\n");
return 0;
}