#900635 (C++17) No.2408 Lakes and Fish

• 問題: No.2408 Lakes and Fish
• ユーザー: zezero
• 提出日時: 2023-08-11 22:08:23
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 173 ms / 2,000 ms
• コード長: 1,196 bytes
• コンパイル時間: 3,661 ms
• コンパイル使用メモリ: 177,584 KB
• 実行使用メモリ: 6,656 KB
• 最終ジャッジ日時: 2024-04-29 13:06:10
• 合計ジャッジ時間: 7,224 ms
• ジャッジサーバーID （参考情報）: judge5 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	156 ms; 6,400 KB
testcase_05	AC	157 ms; 6,400 KB
testcase_06	AC	147 ms; 6,400 KB
testcase_07	AC	173 ms; 6,528 KB
testcase_08	AC	172 ms; 6,656 KB
testcase_09	AC	171 ms; 6,656 KB
testcase_10	AC	172 ms; 6,400 KB
testcase_11	AC	90 ms; 5,376 KB
testcase_12	AC	122 ms; 5,760 KB
testcase_13	AC	34 ms; 5,376 KB
testcase_14	AC	72 ms; 5,376 KB
testcase_15	AC	139 ms; 6,016 KB
testcase_16	AC	49 ms; 5,376 KB
testcase_17	AC	40 ms; 5,376 KB
testcase_18	AC	83 ms; 5,376 KB
testcase_19	AC	132 ms; 5,760 KB
testcase_20	AC	129 ms; 5,888 KB
testcase_21	AC	136 ms; 5,760 KB

ソースコード

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <utility>
#include <map>
#include <set>
#include <queue>
#include <iomanip>
#include <cmath>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long ll;
#define rep(i,n) for (int i = 0; i < int(n);i++)

int main(){
  int n,m;
  cin >> n >> m;
  vector<ll> l(n);
  vector<ll> f(m),b(m),w(m);
  for (int i = 0; i < n;i++){
    cin >> l[i];
  }
  for (int i = 0; i < m;i++){
    cin >> f[i] >> b[i] >> w[i];
  }
  ll ans = 0;
  for (int i = 0; i < m;i++){
    //cout << ans << endl;
    auto itr = lower_bound(l.begin(),l.end(),f[i]);
    if (*itr == f[i]){
      ans += w[i];
      continue;
    }
    else if (itr == l.end()){
      ll diff = abs(f[i]-*(l.rbegin()));
      ans += max({b[i],w[i]-diff});
      continue;
    }
    else if (itr == l.begin()){
      ll diff = abs(f[i]-*(l.begin()));
      ans += max(b[i],w[i]-diff);
      continue;
    }
    else{
      ll diff1 = abs(f[i]-*itr),diff2 = abs(f[i]-*(--itr));
      ll val = b[i];      
      ll val1 = w[i]-diff1,val2 = w[i]-diff2;
      ans += max({val,val1,val2});
    }
  }
  cout << ans << endl;

  return 0;
}