前提・実現したいこと
ドワンゴの挑戦状B問題をc++を使って解きたいです。
方針は、Atcoderのyoutube解説を参考にしました。
作成したコードを提出すると、WAとなってしまいます。入力例1、2では正しい出力が得られます。WAとなってしまう入力はどのようなものが考えられますか?
方針
それぞれの区間を通るスライムの個数の期待値を求め、その期待値とそれぞれの区間の距離を掛けて和を取ったものを出力とします。
左からn番目の区間を通るスライムの個数の期待値yは、
y = 1 + 1/2 + 1/3 + ... + 1/n
と表せます。
発生している問題
入力例1、2では正しい出力が得られるのですが、提出時はそれ以外の入力ではすべてWAとなってしまいます。
該当のソースコード
c++
1#include <bits/stdc++.h> 2using namespace std; 3#define rep(i,n) for(int i=0; i<(n); i++) 4typedef long long ll; 5 6const ll mod = 1000000007; 7 8// 階乗を計算 9ll kaijou(int n) { 10 ll res = 1; 11 for(int i=1; i<=n; i++) { 12 res *= i; 13 } 14 return res; 15} 16 17int main(){ 18 int n; 19 cin >> n; 20 ll x[n]; 21 rep(i, n) cin >> x[i]; 22 23 ll kai = kaijou(n-1); 24 25 // 区間を通るスライムの個数の期待値 26 ll y = 1.0 * kai; 27 28 ll sum = 0; 29 30 rep(i, n-1) { 31 sum += y * (x[i+1]-x[i]); 32 sum %= mod; 33 y += 1.0/(i+2.0) * kai; 34 y %= mod; 35 } 36 37 cout << sum << endl; 38 39 return 0; 40}
試したこと(追記)
階乗の桁溢れを考慮して、kaijou()を編集しました。
c++
1ll kaijou(ll n) 2{ 3 if (n <= 1) 4 return 1; 5 else 6 return (n * kaijou(n - 1)) % mod; 7}
回答2件
あなたの回答
tips
プレビュー