poj 2823 Sliding Window (STL超时版)

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

　　五一假期最大的收获不是学会了什么算法，而是学会了用STL(Standard Template Library)中的set, priority_queue, pair, map等的基本操作。

　　看到priority_queue的时候就让我想起单调队列，想起了poj 2823这题。于是，我就拿这题来做我的STL练习工具，写了一个用STL完成的单调队列。本地测试通过，理论上也是不会出错的，可惜的是，提交上去因为STL的处理方式以及题目的输出格式，使得我每个case的常数太大了，所以最终超时结束。

　　昨晚跟一位师兄讨论STL相关问题的时候，他说了，国内的OJ很多都会卡STL，poj尤其严重，今天总算遇见到了。

　　好吧，我还是上我的STL练习代码吧~虽然是超时了，但是我觉得对我的学习是有帮助的，也因此保存！

　　哪位神牛知道怎么用priority_queue可以不超时的？诚心请教！

#include "cstdio"
#include "vector"
#include "algorithm"
#include "queue"
using namespace std;

struct temp{
    int p;
    int v;
    temp(){
    }
    temp(int pos, int val):
        p(pos), v(val){
    }
    bool operator <(const temp x) const{
        return v < x.v;
    }
    bool operator ==(const temp x) const{
        return v == x.v;
    }
    bool operator >(const temp x) const{
        return v >x.v;
    }
};

int in[1000005];
priority_queue< temp, vector<temp>, less<temp> > T;
priority_queue< temp, vector<temp>, greater<temp> > S;

int main(){
    int n, k;

    while (~scanf("%d%d", &n, &k)){
        temp a;

        for (int i=1; i<=n; i++)
            scanf("%d", &in[i]);
        for (int i=1; i<k; i++){
            a=temp(i, in[i]);
            S.push(a);
        }
        for (int i=k; i<=n; i++){
            a=temp(i, in[i]);
            S.push(a);
            while (S.top().p <= i-k)
                S.pop();
            printf("%d ", S.top().v);
        }
        printf("\n");

        for (int i=1; i<k; i++){
            a=temp(i, in[i]);
            T.push(a);
        }
        for (int i=k; i<=n; i++){
            a=temp(i, in[i]);
            T.push(a);
            while (T.top().p <= i-k)
                T.pop();
            printf("%d ", T.top().v);
        }
        printf("\n");
    }

    return 0;
}

 

p.s.:感谢lzq师兄的指导，感谢PKKJ师兄的STL教程！

如果PKKJ师兄看到，我想问，set, map等中的查找的判断：

　　s.find("content") != s.end()

跟

　　s.count("content")

等价吗？