AtCoder Beginner Contest 188-D题-Snuke Prime-题解

AtCoder Beginner Contest 188-D题-Snuke Prime-题解

题目链接：D - Snuke Prime.

心得体会

本题主要应用了差分，差分的好处就是：只需要改变差分数组首和尾的值即可实现原数组中一段的数据都加上或者减去一个值，时间复杂度大大降低。

选用map容器而不是直接使用数组的好处是：不用记录第一个项目开始时间和最后一个项目结束时间，中间也会少遍历很多次，节省时间和空间。

为了避免数据过大，所以全部采用了long long。

下面附上AC代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
map<ll, ll> s;

int main() {
  ll n, m;
  cin >> n >> m;
  for (ll i = 0; i < n; i++) {
    ll a, b, c;
    cin >> a >> b >> c;
    s[a] += c;
    s[b + 1] -= c;  //差分
  }
  ll ans = 0, price = 0, pos = 0;
  for (auto it = s.begin(); it != s.end(); it++) {
    ans += ((*it).first - pos) * min(price, m);
    price += (*it).second;
    pos = (*it).first;
  }
  cout << ans << endl;
  system("pause");
  return 0;
}