poj 2452 Sticks Problem (rmq+二分，需要返回最值位置)

2016-05-18 · 2 min read · binary search rmq

题意：给你一组数a[n]，求满足a[i] < a[k] < a[j] (i <= k <= j)的最大的j-i。

思路：大概能想到是rmq，然后想出了一个错误复杂度的错误思路，还直到对拍才发现==

收获最大的是：

对于最大值和最小值返回val还是位置的转化竟然可以这样容易!

只要

1int _min(int l,int r)
2{
3    if (a[l]<a[r]) return l;
4    return r;
5}

这样一个函数就可以实现完美转化。。。

/* ***********************************************
Author :111qqz
Created Time :2016年05月16日 星期一 13时42分56秒
File Name :code/poj/2452.cpp
************************************************ */

 1#include <cstdio>
 2#include <cstring>
 3#include <iostream>
 4#include <algorithm>
 5#include <vector>
 6#include <queue>
 7#include <set>
 8#include <map>
 9#include <string>
10#include <cmath>
11#include <cstdlib>
12#include <ctime>
13#define fst first
14#define sec second
15#define lson l,m,rt<<1
16#define rson m+1,r,rt<<1|1
17#define ms(a,x) memset(a,x,sizeof(a))
18typedef long long LL;
19#define pi pair < int ,int >
20#define MP make_pair

 1using namespace std;
 2const double eps = 1E-8;
 3const int dx4[4]={1,0,0,-1};
 4const int dy4[4]={0,-1,1,0};
 5const int inf = 0x3f3f3f3f;
 6const int N=5E4+7;
 7int n;
 8int a[N];
 9int dp[N][20];
10int dp2[N][20];

 1int _min(int l,int r)
 2{
 3    if (a[l]<a[r]) return l;
 4    return r;
 5}
 6int _max( int l,int r)
 7{
 8    if (a[l]>a[r]) return l;
 9    else return r;
10}
11void max_init()
12{
13    for ( int i = 1 ; i <= n ; i++) dp[i][0] = i;

1    for ( int j = 1 ; (1<<j)<=n ; j++)
2	for (int i = 1 ; i + (1<<j)-1 <= n ;i++)
3	    dp[i][j] = _max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
4}

1void min_init()
2{
3    for ( int i = 1 ;i  <= n ; i++) dp2[i][0] = i;

1    for ( int j = 1 ; (1<<j)<= n ; j++)
2	for ( int i = 1 ; i + (1<<j) -1 <= n ; i++)
3	    dp2[i][j] = _min(dp2[i][j-1],dp2[i+(1<<(j-1))][j-1]);
4}

1int rmq_max(int l,int r)
2{
3    int k = 0 ;
4    while (1<<(k+1)<=r-l+1) k++;
5    return _max(dp[l][k],dp[r-(1<<k)+1][k]);
6}

1int rmq_min( int l,int r)
2{
3    int k = 0  ;
4    while (1<<(k+1)<=r-l+1) k++;
5    return _min(dp2[l][k],dp2[r-(1<<k)+1][k]);
6}

 1int bin (int x,int l,int r)
 2{
 3    while (l<=r)
 4    {
 5	if (l==r) return l;
 6	int m = (l+r)>>1;
 7	if (a[x]<a[rmq_min(l,m)])
 8	    l = m + 1;
 9	else r = m;
10    }
11}
12void solve()
13{

 1    int ans = 0;
 2    for ( int i = 1 ; i+ans < n ; i++)
 3    {
 4	int r = bin(i,i+1,n);
 5	int k = rmq_max(i,r);
 6	if (a[k]>a[i])
 7	    ans = max(ans,k-i);
 8    }
 9    if (ans==0) puts("-1");
10    else printf("%d\n",ans);
11}
12int main()
13{
14	#ifndef  ONLINE_JUDGE 
15	freopen("code/in.txt","r",stdin);
16  #endif
17	 while (scanf("%d",&n)!=EOF)
18	{
19	    for ( int i = 1 ; i <= n ; i++) scanf("%d",&a[i]);
20	    max_init();
21	    min_init();

	    solve();
	//    cout<<"sadsad"<<endl;

	}

1  #ifndef ONLINE_JUDGE  
2  fclose(stdin);
3  #endif
4    return 0;
5}