前提・実現したいこと
AtcoderのABC76のD問題について,作成したプログラムがシステムテストを通らない(誤りを含む)理由がわかりません.実行時間が制限時間を超えてしまうTLEが発生しています.
使用言語はC++です.
方針
それぞれの時刻に対する速度の上限値を求め,1つ前の時刻と比較することで加速,等速,減速を場合分けしながら面積を求めることで移動距離の最大値がわかる.
ただし,上限速度が整数値にならない場合があるため,0.5秒ごとに時刻をとることとする.
発生している問題・エラーメッセージ
O(N*T)は高々10^5程度であること,実行結果(実行結果ページ)では正解しているテストケースの実行時間には問題がないことから,おそらくある条件で無限ループが起きていると考えています.
該当のソースコード
C++
1#include "bits/stdc++.h" 2using namespace std; 3#define MOD 1000000007 4#define FOR(i,a,b) for(long long i=(a);i<(b);i++) 5#define REP(i,n) FOR(i,0,n) 6#define ITR(itr,mp) for(auto itr = (mp).begin(); itr != (mp).end(); ++itr) 7#define dump(x) cout << #x << " = " << (x) << endl; 8#define debug(x) cout << #x << " = " << (x) << " (L" << __LINE__ << ")" << " " << __FILE__ << endl; 9typedef long long ll; 10typedef pair<ll,ll> P; 11typedef vector<vector<P>> Graph; 12 13 14int main(){ 15 16 int n; 17 cin >> n; 18 19 vector<double> t(n+2),v(n+2); 20 21 t[0] = v[0] = v[n+1] = 0; 22 REP(i,n){ 23 double T; 24 cin >> T; 25 t[i+1] = t[i] + T; 26 } 27 t[n+1] = t[n]; 28 29 REP(i,n) cin >> v[i+1]; 30 31 vector<double> limit(2*t[n]+2,MOD); //0.5sで区間をとる 32 33 // 速度上限の初期化 34 int k = 0; 35 REP(i,n+2){ 36 while(k <= 2*t[i]){ 37 limit[k] = min(limit[k],v[i]); 38 k++; 39 } 40 k--; 41 } 42 43// REP(i,limit.size()) debug(limit[i]); 44 45 // 範囲の右端からの上限更新 46 REP(i,n+1){ 47 double a = 1; 48 while(limit[t[i]*2+a] > limit[t[i]*2]+(a/2) && t[i+1]-t[i] >= a/2){ 49 limit[t[i]*2+a] = limit[t[i]*2]+(a/2); 50 a++; 51 } 52 } 53 54 // 範囲の左端からの上限更新 55 FOR(i,1,n+2){ 56 double a = 1; 57 while(limit[t[i]*2-a] > limit[t[i]*2]+(a/2) && t[i]-t[i-1] >= a/2){ 58 limit[t[i]*2-a] = limit[t[i]*2]+(a/2); 59 a++; 60 } 61 } 62 63 //REP(i,limit.size()) cout << i << ": " << limit[i] << endl; 64 65 double ans = 0; int i = 0; 66 while(i < limit.size()-2){ 67 double cnt = 0; 68 if(limit[i] == limit[i+1]){ 69 while(limit[i] == limit[i+1]){ 70 cnt++; 71 i++; 72 } 73 ans += cnt/2*limit[i]; 74 // debug(ans); 75 } 76 77 else if(limit[i] == limit[i+1]-0.5){ 78 while(limit[i] == limit[i+1]-0.5){ 79 cnt++; 80 i++; 81 } 82 ans += (cnt/2*cnt/2)/2 + cnt/2*(limit[i]-cnt/2); 83 // debug(ans); 84 } 85 86 else if(limit[i] == limit[i+1]+0.5){ 87 while(limit[i] == limit[i+1]+0.5){ 88 cnt++; 89 i++; 90 } 91 ans += (cnt/2*cnt/2)/2 + cnt/2*limit[i]; 92// debug(ans); 93 } 94 } 95 96 cout << ans << endl; 97 98 return 0; 99 100}
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。