test/yuki1358.test.cpp

View this file on GitHub
Last update: 2021-06-21 15:24:20+09:00
Problem: https://yukicoder.me/problems/no/1358

Depends on

Code

#define PROBLEM "https://yukicoder.me/problems/no/1358"
#include <iostream>
#include <algorithm>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <numeric>
#include <bitset>
#include <cmath>

static const int MOD = 1000000007;
using ll = long long;
using uint = unsigned;
using ull = unsigned long long;
using namespace std;

template<class T> constexpr T INF = ::numeric_limits<T>::max() / 32 * 15 + 208;

#include "../math/extgcd.cpp"
#include "../util/ifloor.cpp"

void solve(){
    ll a, b, c, y;
    cin >> a >> b >> c >> y;
    if(a > c) swap(a, c);
    if(b > c) swap(b, c);
    ll ans = 0, g = gcd(a, b);
    a /= g; b /= g;
    ll p, q;
    extgcd(a, b, p, q);
    for (ll x2 = y; x2 >= 0; x2 -= c) {
        if(x2 % g) continue;
        ll x = x2/g;
        ans += max(0LL, ifloor(q*x, a)-iceil(-p*x, b)+1);
    }
    cout << ans%MOD << "\n";
}

int main() {
    int t;
    cin >> t;
    while(t--){
        solve();
    }
    return 0;
}

#line 1 "test/yuki1358.test.cpp"
#define PROBLEM "https://yukicoder.me/problems/no/1358"
#include <iostream>
#include <algorithm>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <numeric>
#include <bitset>
#include <cmath>

static const int MOD = 1000000007;
using ll = long long;
using uint = unsigned;
using ull = unsigned long long;
using namespace std;

template<class T> constexpr T INF = ::numeric_limits<T>::max() / 32 * 15 + 208;

#line 1 "math/extgcd.cpp"
template<typename T>
T extgcd(T a, T b, T &x ,T &y){
    for (T u = y = 1, v = x = 0; a; ) {
        ll q = b/a;
        swap(x -= q*u, u);
        swap(y -= q*v, v);
        swap(b -= q*a, a);
    }
    return b;
}
 
#line 1 "util/ifloor.cpp"
template<class T>T ifloor(T x, T y){
    return x/y-(x%y ? (x < 0)^(y < 0) : 0);
}

template<class T>inline T iceil(T x, T y){
    return x/y+(x%y ? (x >= 0)^(y < 0) : 0);
}
#line 22 "test/yuki1358.test.cpp"

void solve(){
    ll a, b, c, y;
    cin >> a >> b >> c >> y;
    if(a > c) swap(a, c);
    if(b > c) swap(b, c);
    ll ans = 0, g = gcd(a, b);
    a /= g; b /= g;
    ll p, q;
    extgcd(a, b, p, q);
    for (ll x2 = y; x2 >= 0; x2 -= c) {
        if(x2 % g) continue;
        ll x = x2/g;
        ans += max(0LL, ifloor(q*x, a)-iceil(-p*x, b)+1);
    }
    cout << ans%MOD << "\n";
}

int main() {
    int t;
    cin >> t;
    while(t--){
        solve();
    }
    return 0;
}