IndiaHacks 2016-온라인 에 디 션(Div.1+Div.2)C.곰 과 다운 폭력

C. Bear and Up-Down
제목 연결:
http://www.codeforces.com/contest/653/problem/C
Description
The life goes up and down, just like nice sequences. Sequence t1, t2, ..., tn is called nice if the following two conditions are satisfied:
ti < ti + 1 for each odd i < n; ti > ti + 1 for each even i < n. For example, sequences (2, 8), (1, 5, 1) and (2, 5, 1, 100, 99, 120) are nice, while (1, 1), (1, 2, 3) and (2, 5, 3, 2) are not.
Bear Limak has a sequence of positive integers t1, t2, ..., tn. This sequence is not nice now and Limak wants to fix it by a single swap. He is going to choose two indices i < j and swap elements ti and tj in order to get a nice sequence. Count the number of ways to do so. Two ways are considered different if indices of elements chosen for a swap are different.
Input
The first line of the input contains one integer n (2 ≤ n ≤ 150 000) — the length of the sequence.
The second line contains n integers t1, t2, ..., tn (1 ≤ ti ≤ 150 000) — the initial sequence. It's guaranteed that the given sequence is not nice.
Output
Print the number of ways to swap two elements exactly once in order to get a nice sequence.
Sample Input
5 2 8 4 7 7
Sample Output
2
Hint
제목
하나의 서열 이 나 이 스 로 정의 되면 이 서열 이 계단 모양 을 만족 시 키 는 것 이다.
i 가 짝수 라면 ai>ai-1,ai>ai+1
i 가 홀수 라면 ai지금 은 두 개의 수의 위 치 를 교환 할 수 있 습 니 다.모두 몇 가지 교환 방식 이 있 는 지 물 어보 면 이 서열 을 nice 로 바 꿀 수 있 습 니 다.
처음부터 서열 이 나 이 스 가 아니 라 는 것 을 보증 합 니 다.
문제 풀이:
우리 가 정의 하 는 나 이 스 의 수 는 조건 을 만족 시 키 지 못 하 는 위치 이다.
우 리 는 대담 하 게 한 발 을 추측 할 수 있다.나 이 스 의 수 는 많 지 않 을 것 이다.한 번 의 교환 이 최대 6 개의 수 에 영향 을 미 치기 때문에 우 리 는 이 나 이 스 의 수 를 한 배열 에 직접 던 질 것 이다.
그리고 폭력 은 전체 서열 과 교환 하면 된다.
그리고 check 도 check 의 그 나 쁜 위치 로 당신 과 교환 하 는 그 위치의 수 입 니 다.
코드
#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5+7;
int n;
int a[maxn];
vector<int>tmp;
long long ans = 0;
set<pair<int,int> >S;
bool check()
{
    for(int i=0;i<tmp.size();i++)
    {
        for(int j=-1;j<=1;j++)
        {
            if(tmp[i]+j==0)continue;
            if(tmp[i]+j==n+1)continue;
            int pos = (tmp[i]+j);
            if(pos%2==1)
            {
                if(a[pos]>=a[pos+1])return false;
                if(a[pos]>=a[pos-1])return false;
            }
            else
            {
                if(a[pos]<=a[pos+1])return false;
                if(a[pos]<=a[pos-1])return false;
            }
        }
    }
    return true;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    a[0]=1e9;
    if(n%2==1)a[n+1]=1e9;
    else a[n+1]=-1;
    for(int i=1;i<=n;i++)
    {
        if( i & 1 ){
            bool ok = true;
            if( i + 1 <= n && a[i] >= a[i+1] ) ok = false;
            if( i - 1 >= 1 && a[i] >= a[i-1] ) ok = false;
            if( ok == false ) tmp.push_back( i );
        }else{
            bool ok = true;
            if( i + 1 <= n && a[i] <= a[i+1] ) ok = false;
            if( i - 1 >= 1 && a[i] <= a[i-1] ) ok = false;
            if( ok == false ) tmp.push_back( i );
        }
    }

    if(tmp.size()>30)
        return puts("0"),0;

    for(int i=0;i<tmp.size();i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(tmp[i]==j)continue;
            swap(a[tmp[i]],a[j]);
            bool ok = check();
            if(j%2==1)
            {
                if(a[j]>=a[j+1]||a[j]>=a[j-1])ok=false;
            }
            if(j%2==0)
            {
                if(a[j]<=a[j+1]||a[j]<=a[j-1])ok=false;
            }
            if(ok)
            {
                pair < int , int > SS = make_pair( min( tmp[i] , j ) , max( tmp[i] , j ) );
                if(!S.count(SS)){
                    S.insert( SS );
                    ans++;
                }
            }
            swap(a[tmp[i]],a[j]);
        }
    }
    cout<<ans<<endl;
}

좋은 웹페이지 즐겨찾기