POJ 2823 Sliding Window ST RMQ

 1 #include <stdio.h>
 2 #include <algorithm>
 3 //#define LOG[i] = (i & (i - 1)) ? LOG[i - 1] : LOG[i - 1] + 1
 4 #define MAXX 1234567
 5 #include <vector>
 6 using namespace std;
 7 
 8 int a[MAXX];
 9 int dp1[MAXX][22];
10 int LOG[MAXX];
11 
12 void init(int n)
13 {
14     LOG[1] = 0; 
15     for(int i=2; i<=n; i++)
16         LOG[i]=(i&(i-1))?LOG[i-1]:LOG[i-1]+1;
17 }
18 
19 int ST(int l, int r, int i)
20 {
21     int k=LOG[r-l+1];
22     if(i==1)
23         return max(dp1[l][k],dp1[r-(1<<k)+1][k]);
24     if(i==0)
25         return min(dp1[l][k],dp1[r-(1<<k)+1][k]);
26 }
27 int main()
28 {
29     int n, k;
30     
31     while(~scanf("%d%d",&n, &k))
32     {
33         int i, j;
34         init(n); 
35         for(i=1; i<=n; i++)
36         {
37             scanf("%d", &a[i]);
38             dp1[i][0]=a[i];
39         }
40         for(j=1; j<=20; j++)
41         { 
42             for(i=1; i<=n; i++)
43             {
44                 if(i+(1<<j)-1>n)
45                     break;
46                 dp1[i][j]=min(dp1[i][j-1], dp1[i+(1<<(j-1))][j-1]);
47             }
48         }
49         for(i=1; i<=n-k+1; i++)
50         {
51             if(i!=1)
52                 printf(" ");
53             printf("%d", ST(i,i+k-1,0));
54         }
55         //////
56 
57         for(i=1; i<=n; i++)
58         {
59             dp1[i][0]=a[i];
60             for(j=1; j<=20; j++)
61                 dp1[i][j]=0;
62         }
63         for(j=1; j<=20; j++)
64         { 
65             for(i=1; i<=n; i++)
66             {
67                 if(i+(1<<j)-1>n)
68                     break;
69                 dp1[i][j]=max(dp1[i][j-1], dp1[i+(1<<(j-1))][j-1]);
70             }
71         } 
72         printf("\n");
73         for(i=1; i<=n-k+1; i++)
74         {
75             if(i!=1)
76                 printf(" ");
77             printf("%d", ST(i,i+k-1,1));
78         }
79         printf("\n");
80     }
81 }