【BZOJ2223&&3524】PATULJCI [主席树]

PATULJCI

Time Limit: 10 Sec  Memory Limit: 259 MB
[Submit][Status][Discuss]

Description

Input

　　第一行两个整数n,INF，表示序列长度和ai的上限；
　　第二行有n个数，表示ai；
　　然后有一个整数m，表示询问个数；
　　接下来每行两个l,r，表示询问区间[l,r]中的答案。

Output

　　输出m行，表示对于每个询问的答案。如果有这个数，则输出“yes”，然后输出数的值；否则输出“no”。

Sample Input

　　10 3
　　1 2 1 2 1 2 3 2 3 3
　　8
　　1 2
　　1 3
　　1 4
　　1 5
　　2 5
　　2 6
　　6 9
　　7 10

Sample Output

　　no
　　yes 1
　　no
　　yes 1
　　no
　　yes 2
　　no
　　yes 3

HINT

　　1<=n<=300000 , 1<=m<=10000 , 1<=ai<=10000。

Solution

　　显然是一个主席树，我们建立一棵主席树然后查询是否存在个数>(l+r-1)/2的即可。

Code

 #include<iostream>
 #include<string>
 #include<algorithm>
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<cmath>
 using namespace std;  

 const int ONE=;

 int n,INF,m;
 int x,y,cnt;
 int res_value,res_num;

 struct power
 {
         int root;
         int value;
         int left,right;
 }Node[ONE*];

 int get()
 {
         int res=,Q=;char c;
         while( (c=getchar())< || c> )
         if(c=='-')Q=-;
         res=c-;
         while( (c=getchar())>= && c<= )
         res=res*+c-;
         return res*Q;
 }

 void Update(int &x,int y,int L,int R,int Q)
 {
         x = ++cnt;
         Node[x].left = Node[y].left;
         Node[x].right = Node[y].right;
         Node[x].value = Node[y].value + ;
         if(L == R) return;

         int M = (L+R)>>;
         if(Q <= M)
             Update(Node[x].left,Node[y].left, L,M, Q);
         else
             Update(Node[x].right,Node[y].right, M+,R, Q);
 }

 void Query(int x,int y,int L,int R,int Kth)
 {
         if(L == R)
         {
             res_value = L;
             res_num = Node[y].value - Node[x].value;
             return;
         }

         int M = (L+R)>>;
         int record = Node[Node[y].left].value - Node[Node[x].left].value;

         if(Kth < record)
             Query(Node[x].left,Node[y].left, L,M, Kth);
         else
             Query(Node[x].right,Node[y].right, M+,R, Kth);
 }

 int main()
 {
         n=get();    INF=get();
         for(int i=;i<=n;i++)
         {
             x=get();
             Update(Node[i].root,Node[i-].root, ,INF, x);
         }

         m=get();
         for(int i=;i<=m;i++)
         {
             x=get();    y=get();

             res_value = , res_num = ;
             int M = (y-x+)/;
             Query(Node[x-].root,Node[y].root, ,INF, M);

             if(res_num > M)
                 printf("yes %d",res_value);
             else
                 printf("no");
             printf("\n");
         }
 }