HDU YY's new problem

시간 이 많이 걸 리 는 문 제 를 만 났 습 니 다. AC 를 제출 한 후에 도 가끔 지나 갈 수 있 고 가끔 지나 갈 수 없습니다. 우연히 지나 가 는 것 은 정말 쉽 지 않 지만 본 문제 의 사고 와 생각 은 설명 이 필요 합 니 다.
만약 에 순 폭력 적 인 방법 으로 서열 중의 각 숫자 를 순서대로 옮 겨 다 니 면 복잡 도 는 O (n3) 이 고 반드시 시간 을 초과 해 야 한다. 반드시 알고리즘 을 최적화 시 켜 야 한다. 이 문제 의 특수성 을 고려 하면 서열 에는 n 개의 숫자 만 있 고 1 에서 n 까지 만족 해 야 하 는 것 은 a - b = b - c (그리고 a, b, c 순 서 는 바 꿀 수 없다) 와 같은 특징 을 가 진 식 으로 a + c = 2 * b 로 바 꿀 수 있다.(또한 a, b, c 순 서 는 바 꿀 수 없습니다), 즉 a + b 는 반드시 짝수 이 고 아래 표 시 를 기록 하 는 방법 을 사용 해 야 합 니 다. (a + c) / 2 의 값 에 대응 하 는 아래 표 시 는 a 의 아래 표 와 c 의 아래 표 사이 에 있 으 면 이러한 관 계 를 만족 시 키 는 순 서 를 찾 을 수 있 습 니 다. 실질 적 으로 Hash 표를 간단하게 활용 하 였 습 니 다.
HDU YY's new problem http://acm.hdu.edu.cn/showproblem.php?pid=3833
코드 구현 은 다음 과 같 습 니 다.

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

int a[10010];
int b[10010];

int main()
{
    int t,n,sum;
    int flag;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        flag=0;
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            b[a[i]]=i;   //    
        }
        for(int i=1; i<=n; i++)
        {
            for(int j=i+1; j<=n; j++)
            {
                sum=a[i]+a[j];
                if(sum%2==1)
                    continue;
                else if(sum%2==0)
                {
                    if((b[sum/2] > b[a[i]])&&(b[sum/2] < b[a[j]]))    //if((b[sum/2] > i)&&(b[sum/2] < j)
                    {
                        flag=1;
                        break;
                    }
                }
            }
            if(flag)
                break;
        }
        if(flag==1)
            printf("Y
");
        else printf("N
");
    }
    return 0;
}