C ++ Stl中的优先级队列

Question

我想设计一个优先级队列，该队列在t []值较低的元素的优先级最高。 这件事我能实现。

但是我也希望s []值为零的元素应该在末尾（不管它们的t []值如何）我该如何修改我的代码？

#include <bits/stdc++.h>
using namespace std;
#define ll long long

ll t[4];
ll s[4];

struct func
{
  bool operator()(const ll lhs, const ll rhs) const
  {
    return(t[lhs] > t[rhs]);
  }
};

int main(){
t[0]=2;t[1]=3;t[2]=0;t[3]=6;
s[0]=0;s[1]=0;s[2]=1;s[3]=1;
priority_queue<ll,vector<ll>,func>pq;

pq.push(0);
pq.push(1);
pq.push(2);
pq.push(3);

// for displaying

cout<<pq.top()<<endl;
pq.pop();

cout<<pq.top()<<endl;
pq.pop();

cout<<pq.top()<<endl;
pq.pop();

cout<<pq.top()<<endl;
pq.pop();
}

Answer 1

两者都检查：如果s之一为零，则将其作为较低优先级返回，否则比较t s：

bool operator() (const ll lhs, const ll rhs) const {
  const bool leftIsZero = s[lhs] == 0;
  const bool rightIsZero = s[rhs] == 0;
  const bool oneIsZero = leftIsZero ^ rightIsZero;

  if (oneIsZero)
    return rightIsZero;

  return t[lhs] > t[rhs];
}