解決したい問題の内容
言語はC++です。AtCoderの問題アルゴリズム検定過去問G問題を解いていました。
https://atcoder.jp/contests/past201912-open/tasks/past201912_g
↑このリンク先の問題です。
再帰関数を用いてコードして提出しました。
しかし、コンパイルは通ったもののWAの結果が出ました。
WAの原因とコードの改善点を知りたいです。
発生している問題・エラーメッセージ
下のリンク先のテストケースを見ていただくとわかる通り、ばらばらにACとWAが出ました。
該当のソースコード
C++
1#include <iostream> 2using namespace std; 3 4void happiness(int cnt); 5int** pHappiness; 6int* pGroup; 7long long* pHappy; 8long long ans; 9int N; 10 11void Happiness(int cnt){ 12 int preHappy; 13 for(int i = 0; i < 3; ++i){ 14 pGroup[cnt] = i; 15 if(cnt == 0) *pHappy = 0; 16 if(i == 0) preHappy = 0; 17 else *pHappy = *pHappy - preHappy; 18 19 for(int j = 0; j < cnt; ++j){ 20 if(pGroup[j] == pGroup[cnt]) preHappy += pHappiness[j][cnt-1]; 21 } 22 *pHappy += preHappy; 23 24 if(cnt < N-1){ 25 Happiness(cnt+1); 26 }else{ 27 if(ans < *pHappy) ans = *pHappy; 28 } 29 } 30 *pHappy -= preHappy; 31 return; 32} 33 34int main(void){ 35 cin >> N; 36 pHappiness = new int*[N-1]; 37 for(int i = 0; i < N-1; ++i){ 38 pHappiness[i] = new int[N-1]; 39 //pHappiness[i][j] = Happiness between i and i+j+1 40 for(int j = 0; j+i < N-1; ++j) cin >> pHappiness[i][i+j]; 41 } 42 43 int group[N]; 44 45 pHappy = new long long; 46 pGroup = group; 47 ans = -2147483648; 48 Happiness(0); 49 50 cout << ans << endl; 51}
最終的にこのようなコードで提出しました。
https://atcoder.jp/contests/past201912-open/submissions/13699538
↑問題のコードのリンクです。提出結果もこちらです。
試したこと
いくつかACが存在するので、入力された値の特殊な場合をはじきだせてないのが原因だと思います。
・変数のオーバーフローについては、余裕をもって大きい値はlong longに入れてあるはずなので、ないと思います。実際すべてのペアの幸福度を1,000,000にしてやってみましたが正常作動でした。
・幸福度を-1,000,000や0にした場合についてもうまくいっていました
・幸福度を正の値、負の値交互に入れて見たりしました。うまくいっていました
他のACを獲得しておられる方のコードと、上記の入力を試して出力を比較してみたりしたのですが、結局原因がわかりません。
補足情報(FW/ツールのバージョンなど)
テストはAtCoderに組み込まれているコードテスト機能を使っています。
私事ですが、今回初めてteratailで質問します。
質問の仕方、こうしたほうがいいよ!
というアドバイスがありましたら、そちらもぜひ教えていただきたいです。
よろしくお願いします。