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;
}