Black Box

http://poj.org/problem?id=1442

 1 #include<cstdio>
 2 #include<algorithm>
 3 #include<queue>
 4 #define MAXN 30010
 5 long long a[MAXN],b[MAXN];
 6 using namespace std;
 7 int main()
 8 {
 9     priority_queue<long long ,vector<long long >,greater<long long> >q;  //从小到大排列  小根堆
10     priority_queue<long long ,vector<long long>,less<long long> >p;// 从大到小排列 大根堆
11     int n,m;
12     long long t;
13     scanf("%d%d",&n,&m);
14     for(int i=0;i<n;i++)
15         scanf("%lld",&a[i]);
16     for(int i=0;i<m;i++)
17         scanf("%lld",&b[i]);
18     int tt=0;
19     for(int j=0;j<m;j++)
20     {
21         for( int i=tt;i<b[j];i++) 
22         {
23             q.push(a[i]);
24             if(!p.empty()&&q.top()<p.top())  //对每一步add之后操作
25             {
26                 t=q.top();
27                 q.pop();
28                 q.push(p.top());
29                 p.pop();
30                 p.push(t);
31             }
32         }
33         tt=b[j];
34         printf("%lld\n",q.top());
35         p.push(q.top());
36         q.pop();
37     }
38     return 0;
39 }
View Code

 另外的做法，这是我用堆写的。

  1 #include<cstdio>
  2 #include<algorithm>
  3 #include<cstring>
  4 #define MAXN 60010
  5 long long a[MAXN],b[MAXN],c[MAXN],d[MAXN];
  6 using namespace std;
  7 int len=0,len1=0;
  8 void up1(int n)
  9 {
 10     a[++len]=n;
 11     int p=len;
 12     int q=p/2;
 13     long long m=a[p];
 14     while(q>=1&&m<a[q])
 15     {
 16         a[p]=a[q];
 17         p=q;
 18         q=p/2;
 19     }
 20     a[p]=m;
 21 }
 22 void up2(int n)
 23 {
 24     b[++len1]=n;
 25     int p=len1;
 26     int q=p/2;
 27     long long m=b[p];
 28     while(q>=1&&m>b[q])
 29     {
 30         b[p]=b[q];
 31         p=q;
 32         q=p/2;
 33     }
 34     b[p]=m;
 35 }
 36 void down1(int p)
 37 {
 38     a[1]=a[len--];
 39     int q=p*2;
 40     long long m=a[p];
 41     while(q<=len)
 42     {
 43         if(q<len&&a[q]>a[q+1])
 44             q++;
 45         if(a[q]>=m) break;
 46         else
 47         {
 48             a[p]=a[q];
 49             p=q;
 50             q=p*2;
 51         }
 52     }
 53     a[p]=m;
 54 }
 55 int pop1()
 56 {
 57     long long r=a[1];
 58     return r;
 59 }
 60 int pop2()
 61 {
 62     long long  r=b[1];
 63     return r;
 64 }
 65 void down2(int p)
 66 {
 67     b[1]=b[len1--];
 68     int q=p*2;
 69     long long m=b[p];
 70     while(q<=len1)
 71     {
 72         if(q<len1&&b[q]<b[q+1])
 73             q++;
 74         if(b[q]<=m) break;
 75         else
 76         {
 77             b[p]=b[q];
 78             p=q;
 79             q=p*2;
 80         }
 81     }
 82     b[p]=m;
 83 }
 84 
 85 int main()
 86 {
 87     int n,m;
 88     memset(a,0,sizeof(a));
 89     memset(b,0,sizeof(b));
 90     scanf("%d%d",&n,&m);
 91     for(int i=0; i<n; i++)
 92         scanf("%lld",&c[i]);
 93     for(int j=0; j<m; j++)
 94         scanf("%lld",&d[j]);
 95     int tt=0;
 96     for(int i=0; i<m; i++)
 97     {
 98         for(int j=tt; j<d[i]; j++)
 99         {
100             up1(c[j]);
101             if(len1!=0&&a[1]<b[1])
102             {
103                 int t=pop1();
104                 down1(1);
105                 up1(pop2());
106                 down2(1);
107                 up2(t);
108             }
109         }
110         tt=d[i];
111         printf("%lld\n",a[1]);
112         up2(pop1());
113         down1(1);
114     }
115     return 0;
116 }
View Code

 

posted @ 2013-08-04 17:22  null1019  阅读(156)  评论(0)  编辑  收藏  举报