fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. const int MOD = 1e9 + 7;
  4. int r, g;
  5. int dp[200005];
  6.  
  7. main() {
  8.     ios_base::sync_with_stdio(false);
  9.     cin.tie(0); cout.tie(0);
  10.     freopen("TEST.inp", "r", stdin);
  11.     freopen("TEST.out", "w", stdout);
  12.     cin >> r >> g;
  13.  
  14.     int h = sqrt(2 * (r + g));
  15.     int sum = h * (h + 1) / 2;
  16.     if (sum > r + g) {
  17.         h--;
  18.         sum -= (h + 1);
  19.     }
  20.     dp[0] = 1;
  21.     for (int val = 1; val <= h; val++) {
  22.         for (int i = r; i >= val; i--) {
  23.             dp[i] += dp[i - val];
  24.             if (dp[i] >= MOD) dp[i] -= MOD;
  25.         }
  26.     }
  27.     int ans = 0;
  28.     for (int i = max(0, sum - g); i <= r; i++) {
  29.         ans += dp[i];
  30.         if (ans >= MOD) ans -= MOD;
  31.     }
  32.  
  33.     cout << ans << '\n';
  34.  
  35.     return 0;
  36. }
  37.  
Success #stdin #stdout 0s 5320KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
Standard output is empty
https://ideone.com/NWvaMT
language:
C++14 (gcc 8.3)
created:
14 hours ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!