poj 3368 RMQ(查询给定区间内频率最多的数字数量)

来源:互联网 发布:知乎上海生死劫电影 编辑:程序博客网 时间:2024/06/05 19:04

题意:给出n个数的非递减序列和Q个询问(l,r),对于每个询问求出(l,r)之间连续出现次数最多的次数。

思路:用RMQ。将原序列稍作转换。end数组存放与第i个数相同的最后一个数的下标(查找的时候保证O(1))。

#include <stdio.h>#include <string.h>#include <math.h>#define max(a,b) ((a)>(b)?(a):(b))#define N 100010int dp[N][20],s[N],t[N],end[N];int n,q;void st(){    int i,j,k=log((double)(n+1))/log(2.0);    for(i = 1;i<=n;i++)        dp[i][0] = t[i];    for(j = 1;j<=k;j++)        for(i = 1;i+(1<<j)-1 <= n;i++)            dp[i][j] = max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);}int query(int a,int b){    int k = log((double)b-a+1)/log(2.0);    return max(dp[a][k],dp[b-(1<<k)+1][k]);}int main(){    while(scanf("%d",&n) && n){        int i,j,a,b;        memset(t,0,sizeof(t));        scanf("%d",&q);        for(i = 1;i<=n;i++)            scanf("%d",&s[i]);        end[n] = n;        for(i = n-1;i>=1;i--)            end[i] = (s[i]==s[i+1])?end[i+1]:i;        for(t[1] = 1,i = 2;i<=n;i++)            if(s[i] == s[i-1])                t[i] = t[i-1]+1;            else                t[i]++;        st();        while(q--){            scanf("%d %d",&a,&b);            if(end[a] >= b)//表示查询区间只有一种数字                j = b-a+1;            else{                j = end[a]-a+1;                j = max(j,query(end[a]+1,b));            }            printf("%d\n",j);        }    }    return 0;}


0 0
原创粉丝点击