HDU 4334 尺取+思维
题意:在5个数组中,每个数组中取一个数,5个数之和是否为0;
第一次我是用map标记前3个数组的和,然后暴力跑后两个数组的循环,果然T了。。
网上又两种做法:
① 处理前三个数组的和,后两个数组的和,排序+枚举+二分。我自己没试过不知道能过不
t *(n^3+n^2+log(n^3)+log(n^2)+ n^2*log(n^3))这是复杂度,我自己感觉会T.
② 尺取法:处理 1和2 数组,处理 3和4 数组 ,排序+取尺法。
设预处数组 x (1,2和),y(3,4和),i=1,j=n*n,x=-num[5][sign] (sign=1-n)
如果x[i]+y[j]>x 则j--; 如果x[i]+y[j]/<x 则i++;
第二种方法代码:
///#include<bits/stdc++.h>
///#include<unordered_map>
#include<unordered_set>
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<set>
#include<stack>
#include<map>
#include<new>
#include<vector>
#define MT(a,b) memset(a,b,sizeof(a));
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const double E=2.718281828459;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
ll num[6][205],x[40005],y[40005];
int t,n,flag;
bool judge(ll s)
{
int i=1,j=n*n;
while(i<=n*n&&j>=1)
{
if(x[i]+y[j]==s)
return 1;
else if(x[i]+y[j]>s)
j--;
else
i++;
}
return 0;
}
int main()
{
scanf("%d",&t);
while(t--)
{
flag=0;
scanf("%d",&n);
for(int i=1;i<=5;i++)
for(int j=1;j<=n;j++)
scanf("%lld",&num[i][j]);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
x[(i-1)*n+j]=num[1][i]+num[2][j];
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
y[(i-1)*n+j]=num[3][i]+num[4][j];
sort(x+1,x+1+n*n);
sort(y+1,y+1+n*n);
for(int i=1;i<=n&&flag==0;i++)
{
if(judge(-num[5][i]))
flag=1;
}
if(flag)
printf("Yes\n");
else
printf("No\n");
}
return 0;
}