SSTF算法

最近碰到一个专门制作大厂真题模拟题的网站，最近一直在上面刷题。今天来每日一题，这是今年美团0429的一道题。稍微整理了一下自己的思路。下面是题目:

、

这题的评测地址: **https://codefun2000.com/p/P1269**

学一个新东西

multiset

里面是排好序的

可以存重复的值

但是里面的值不能被修改 否则就不能排序了

可以用lower_bound(x)，得到multiset中大于等于x的最小的数的位置

auto ri = q.lower_bound(x)

若x比multiset中的任何数字都要大，则会返回q.end()

用*ri索引到位置上的值

upper_bound(x)则是返回大于x的最小的数的位置 其他一样的

#include <iostream>
#include <set>
#include <algorithm>
using namespace std;
#define int long long
const int N = 100010;
typedef pair<int, int> PII;
int n, m, k;
PII tasks[N];

signed main() {
  ios::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  cin >> n >> m >> k;
  for(int i = 1; i <= n; i++) {
    cin >> tasks[i].second; // position
  }
  for(int i = 1; i <= n; i++) {
    cin >> tasks[i].first; // time
  }
  sort(tasks + 1, tasks + n + 1); // order by time
  int rest_tasks = n;
  int now_pos = k;
  int now_it = 1;
  int now_time = 0; // record current time
  multiset<int> q;
  int res = 0;
  while(rest_tasks--) {
    bool wait = true;
    while(now_it <= n && tasks[now_it].first <= now_time) {
      q.insert(tasks[now_it++].second);
      wait = false;
    }
    if(wait == true && q.size() == 0) {
      now_time = tasks[now_it].first;
      q.insert(tasks[now_it++].second);
    }
    int nextpos = -1;
    int min_dist = 1e9;
    auto ri = q.lower_bound(now_pos);
    auto rm = ri;
    if(ri != q.end()) { // if exist x that >= now_pos
      min_dist = min(min_dist, *ri - now_pos);
      nextpos = *ri;
    }
    if(ri != q.begin()) {
      auto le = --ri;
       if(min_dist >= (now_pos - *le)) {
         min_dist = now_pos - *le;
         nextpos = *le;
         rm = le;
       }
    }
    q.erase(rm);
    res += min_dist;
    now_time += m * min_dist;
    now_pos = nextpos;
  }  
  cout << res << '\n';
  return 0;
}