poj 2442 Sequence (Priority Queue)

2442 -- Sequence

　　真郁闷，明明方法是对的，为什么我的代码老是那么的慢。_(:з」∠)_

　　这题要想考虑两列的情况，然后逐列拓展。

代码如下：

 #include <cstdio>
 #include <cstring>
 #include <iostream>
 #include <algorithm>
 #include <queue>
 
 using namespace std;
 
 const int N = ;
 const int Q = N * N;
 template<class T>
 struct PriQ {
     T q[Q];
     int sz;
     void clear() { sz = ;}
     void up(int k) {
         while (k > ) {
             if (q[k] < q[k >> ]) swap(q[k >> ], q[k]);
             else return ;
             k >>= ;
         }
     }
     void push(T x) {
         q[++sz] = x;
         up(sz);
     }
     void down(int k) {
         while ((k << ) <= sz) {
             if ((k << ) == sz || q[k << ] < q[k <<  | ]) {
                 if (q[k << ] < q[k]) swap(q[k], q[k << ]);
                 else return ;
                 k = k << ;
             } else {
                 if (q[k <<  | ] < q[k]) swap(q[k], q[k <<  | ]);
                 else return ;
                 k = k <<  | ;
             }
         }
     }
     void pop() {
         q[] = q[sz];
         sz--;
         down();
     }
     T top() { return q[];}
     int size() { return sz;}
 } ;
 PriQ<int> pq;
 int ans[N];
 //priority_queue<Node> pq;
 
 template<class T>
 void priqcon(T *a, PriQ<T> &b, int sz) {
     a[] = ;
     for (int i = ; i <= sz; i++) {
         a[++a[]] = b.top();
         b.pop();
     }
 }
 
 int rec[N];
 
 int main() {
     int T, n, m, x;
     scanf("%d", &T);
     while (T-- && ~scanf("%d%d", &n, &m)) {
         ans[] = ;
         ans[] = ;
         int mx;
         for (int i = ; i < n; i++) {
             pq.clear();
             mx = ;
             for (int j = ; j < m; j++) scanf("%d", &rec[j]);
             sort(rec, rec + m);
             for (int j = ; j < m; j++) {
                 x = rec[j];
                 for (int k = ; k <= ans[]; k++) {
                     if (pq.sz < m || ans[k] + x < mx) {
                         pq.push(ans[k] + x);
                         mx = max(mx, ans[k] + x);
                     }
                 }
             }
             priqcon(ans, pq, m);
         }
         for (int i = ; i < m; i++) {
             if (i) putchar(' ');
             printf("%d", ans[i + ]);
         }
         puts("");
     }
     return ;
 }

　　好了，其实做这题并不是我的原意，实际上是为了测试二叉堆有没写错的。

——written by Lyon