poj 2442 Sequence (Priority Queue)

2442 -- Sequence

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

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

代码如下：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

const int N = 2222;
const int Q = N * N;
template<class T>
struct PriQ {
    T q[Q];
    int sz;
    void clear() { sz = 0;}
    void up(int k) {
        while (k > 1) {
            if (q[k] < q[k >> 1]) swap(q[k >> 1], q[k]);
            else return ;
            k >>= 1;
        }
    }
    void push(T x) {
        q[++sz] = x;
        up(sz);
    }
    void down(int k) {
        while ((k << 1) <= sz) {
            if ((k << 1) == sz || q[k << 1] < q[k << 1 | 1]) {
                if (q[k << 1] < q[k]) swap(q[k], q[k << 1]);
                else return ;
                k = k << 1;
            } else {
                if (q[k << 1 | 1] < q[k]) swap(q[k], q[k << 1 | 1]);
                else return ;
                k = k << 1 | 1;
            }
        }
    }
    void pop() {
        q[1] = q[sz];
        sz--;
        down(1);
    }
    T top() { return q[1];}
    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[0] = 0;
    for (int i = 1; i <= sz; i++) {
        a[++a[0]] = 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[0] = 1;
        ans[1] = 0;
        int mx;
        for (int i = 0; i < n; i++) {
            pq.clear();
            mx = 0;
            for (int j = 0; j < m; j++) scanf("%d", &rec[j]);
            sort(rec, rec + m);
            for (int j = 0; j < m; j++) {
                x = rec[j];
                for (int k = 1; k <= ans[0]; 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 = 0; i < m; i++) {
            if (i) putchar(' ');
            printf("%d", ans[i + 1]);
        }
        puts("");
    }
    return 0;
}

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

——written by Lyon