#381640 (C++17) No.890 移調の限られた旋法

提出ソース

結果

問題	No.890 移調の限られた旋法
ユーザー	mai
提出日時	2019-09-20 22:28:01
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	9,484 bytes
コンパイル時間	2,813 ms
コンパイル使用メモリ	213,800 KB
実行使用メモリ	16,320 KB
最終ジャッジ日時	2024-09-14 18:20:39
合計ジャッジ時間	19,356 ms
ジャッジサーバーID （参考情報）	judge4 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	780 ms 16,320 KB
testcase_01	AC	780 ms 9,256 KB
testcase_02	AC	780 ms 9,384 KB
testcase_03	AC	780 ms 9,384 KB
testcase_04	AC	784 ms 9,156 KB
testcase_05	AC	782 ms 9,256 KB
testcase_06	WA	-
testcase_07	AC	779 ms 9,384 KB
testcase_08	AC	780 ms 9,244 KB
testcase_09	AC	779 ms 9,256 KB
testcase_10	AC	779 ms 9,256 KB
testcase_11	AC	781 ms 9,288 KB
testcase_12	AC	779 ms 9,384 KB
testcase_13	TLE	-
testcase_14	TLE	-
testcase_15	--	-
testcase_16	--	-
testcase_17	--	-
testcase_18	--	-
testcase_19	--	-
testcase_20	--	-
testcase_21	--	-
testcase_22	--	-
testcase_23	--	-
testcase_24	--	-
testcase_25	--	-
testcase_26	--	-
testcase_27	--	-
testcase_28	--	-
testcase_29	--	-
testcase_30	--	-
testcase_31	--	-
testcase_32	--	-
testcase_33	--	-
testcase_34	--	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#pragma GCC optimize ("O3")
#include "bits/stdc++.h"

using namespace std;
using ll = long long int;

#define debugos cout
#define debug(v) {printf("L%d %s > ",__LINE__,#v);debugos<<(v)<<endl;}
#define debugv(v) {printf("L%d %s > ",__LINE__,#v);for(auto e:(v)){debugos<<e<<" ";}debugos<<endl;}
#define debuga(m,w) {printf("L%d %s > ",__LINE__,#m);for(int x=0;x<(w);x++){debugos<<(m)[x]<<" ";}debugos<<endl;}
#define debugaa(m,h,w) {printf("L%d %s >\n",__LINE__,#m);for(int y=0;y<(h);y++){for(int x=0;x<(w);x++){debugos<<(m)[y][x]<<" ";}debugos<<endl;}}
#define all(v) (v).begin(),(v).end()
#define repeat(cnt,l) for(auto cnt=remove_reference<decltype(l)>::type();(cnt)<(l);++(cnt))
#define rrepeat(cnt,l) for(auto cnt=(l)-1;0<=(cnt);--(cnt))
#define iterate(cnt,b,e) for(auto cnt=(b);(cnt)!=(e);++(cnt))
#define diterate(cnt,b,e) for(auto cnt=(b);(cnt)!=(e);--(cnt))
const ll MD = 1000000007ll; const long double PI = 3.1415926535897932384626433832795L;
inline void assert_call(bool assertion, function<void()> f) { if (!assertion) { cerr << "assertion fault:" << endl; f(); abort(); } }
template<typename T1, typename T2> inline ostream& operator <<(ostream &o, const pair<T1, T2> p) { o << '(' << p.first << ':' << p.second << ')'; return o; }
template<typename Vec> inline ostream& _ostream_vecprint(ostream& os, const Vec& a) {
    os << '['; for (const auto& e : a) os << ' ' << e << ' '; os << ']'; return os; }
template<typename T> inline ostream& operator<<(ostream& o, const vector<T>& v) { return _ostream_vecprint(o, v); }
template<typename T, size_t S> inline ostream& operator<<(ostream& o, const array<T, S>& v) { return _ostream_vecprint(o, v); }
template<typename T> inline T& chmax(T& to, const T& val) { return to = max(to, val); }
template<typename T> inline T& chmin(T& to, const T& val) { return to = min(to, val); }
void bye(string s, int code = 0) { cout << s << endl; exit(code); }
mt19937_64 randdev(8901016);
template<typename T, typename Random = decltype(randdev), typename enable_if<is_integral<T>::value>::type* = nullptr>
inline T rand(T l, T h, Random& rand = randdev) { return uniform_int_distribution<T>(l, h)(rand); }
template<typename T, typename Random = decltype(randdev), typename enable_if<is_floating_point<T>::value>::type* = nullptr>
inline T rand(T l, T h, Random& rand = randdev) { return uniform_real_distribution<T>(l, h)(rand); }

#if defined(_WIN32) || defined(_WIN64)
#define getc_x _getc_nolock
#define putc_x _putc_nolock
#elif defined(__GNUC__)
#define getc_x getc_unlocked
#define putc_x putc_unlocked
#else
#define getc_x getc
#define putc_x putc
#endif
#define isvisiblechar(c) (0x21<=(c)&&(c)<=0x7E)
class MaiScanner {
    FILE* fp_;
public:
    inline MaiScanner(FILE* fp):fp_(fp){}
    template<typename T> void input_integer(T& var) noexcept {
        var = 0; T sign = 1;
        int cc = getc_x(fp_);
        for (; cc < '0' || '9' < cc; cc = getc_x(fp_))
            if (cc == '-') sign = -1;
        for (; '0' <= cc && cc <= '9'; cc = getc_x(fp_))
            var = (var << 3) + (var << 1) + cc - '0';
        var = var * sign;
    }
    inline int c() noexcept { return getc_x(fp_); }
    inline MaiScanner& operator>>(int& var) noexcept { input_integer<int>(var); return *this; }
    inline MaiScanner& operator>>(long long& var) noexcept { input_integer<long long>(var); return *this; }
    inline MaiScanner& operator>>(string& var) {
        int cc = getc_x(fp_);
        for (; !isvisiblechar(cc); cc = getc_x(fp_));
        for (; isvisiblechar(cc); cc = getc_x(fp_))
            var.push_back(cc);
        return *this;
    }
    template<typename IT> inline void in(IT begin, IT end) { for (auto it = begin; it != end; ++it) *this >> *it; }
};
class MaiPrinter {
    FILE* fp_;
public:
    inline MaiPrinter(FILE* fp):fp_(fp){}
    template<typename T>
    void output_integer(T var) noexcept {
        if (var == 0) { putc_x('0', fp_); return; }
        if (var < 0)
            putc_x('-', fp_),
            var = -var;
        char stack[32]; int stack_p = 0;
        while (var)
            stack[stack_p++] = '0' + (var % 10),
            var /= 10;
        while (stack_p)
            putc_x(stack[--stack_p], fp_);
    }
    inline MaiPrinter& operator<<(char c) noexcept { putc_x(c, fp_); return *this; }
    inline MaiPrinter& operator<<(int var) noexcept { output_integer<int>(var); return *this; }
    inline MaiPrinter& operator<<(long long var) noexcept { output_integer<long long>(var); return *this; }
    inline MaiPrinter& operator<<(char* str_p) noexcept { while (*str_p) putc_x(*(str_p++), fp_); return *this; }
    inline MaiPrinter& operator<<(const string& str) {
        const char* p = str.c_str();
        const char* l = p + str.size();
        while (p < l) putc_x(*p++, fp_);
        return *this;
    }
    template<typename IT> void join(IT begin, IT end, char sep = ' ') { for (bool b = 0; begin != end; ++begin, b = 1) b ? *this << sep << *begin : *this << *begin; }
};
MaiScanner scanner(stdin);
MaiPrinter printer(stdout);

template<int MaxN, long long Mod>
class Combinatorics {
    using value_type = long long;
    value_type fact_[MaxN + 1];
    // value_type inv_[MaxN + 1]; // iranai

public:
    value_type pow(value_type x, value_type p) {
        value_type y = 1; x = x % Mod;
        for (; 0 < p; p >>= 1) {
            if (p & 1) y = y * x % Mod;
            x = x * x % Mod;
        }
        return y;
    }

    Combinatorics() :fact_() {
        fact_[0] = 1;
        fact_[1] = 1;
        for (value_type n = 2; n <= MaxN; ++n)
            fact_[n] = (fact_[n - 1] * n) % Mod;
        // inv_[Max] := calculate;
        // iterate => inv_[i] = inv_[i+1] * (i+1) % MD;
    }
    inline value_type fact(int n) const {
        return fact_[n];
    }

    inline value_type inv(value_type n) {
        return pow(n, MD - 2);
    }

    inline value_type nPr(value_type n, value_type r) {
        if (r < 0 || n < r) return 0;
        return fact_[n] * inv(fact_[n - r]) % MD;
    }
    inline value_type nCr(value_type n, value_type r) {
        if (n < r) return 0;
        return ((fact_[n] * inv(fact_[n - r]) % MD) * inv(fact_[r])) % MD;
    }
    inline value_type nHr(value_type n, value_type r) {
        if (n == 0 && r == 0) return 1;
        if (n == 0) return 0;
        return nCr(n - 1 + r, n - 1);
    }
};

template<int Max = 2000>
class IsPrimeC {
    bool d_[Max + 1];

    template<typename T>
    static constexpr T cesqrt(T s) {
        double x = s / 2.0;
        double prev = 0.0;
        while (x != prev) {
            prev = x;
            x = (x + s / x) / 2.0;
        }
        return x;
    }

public:
    constexpr IsPrimeC(): d_() {
        d_[0] = d_[1] = 1;
        int sqh = cesqrt<int>(Max);
        for (int i = 2; i <= sqh; i++) {
            if (d_[i] == 0) {
                for (int j = i * i; j <= Max; j += i) { // i*i
                    d_[j] = 1;
                }
            }
        }
        for (int i = 0; i <= Max; ++i) d_[i] = !d_[i];
    }
    constexpr inline bool operator[](int x) const{return d_[x]; }

};


template<int Max = 2000>
class PrimeListC {
    int d_[Max];
public:
    PrimeListC() : d_() {
        int n = 1;
        d_[0] = 2;
        for (int x = 3; n < Max; ++x) {
            bool f = true;
            for (int i = 0; d_[i] * d_[i] <= x; ++i)
                if (x / d_[i] * d_[i] == x) { f = false; break; }
            if (f) d_[n++] = x;
        }
    }
    inline int operator[](int x) const { return d_[x]; }
    template<int _Max>
    class iterator {
        const PrimeListC<_Max>& pl;
        int ptr = 0;
    public:
        constexpr iterator(const decltype(pl)& _pl, int _ptr = 0) :pl(_pl),ptr(_ptr) { }
        constexpr int operator*() const { return pl[ptr]; }
        constexpr iterator<_Max>& operator++() { ptr++; return *this; } // prefix
        constexpr inline bool operator!=(const iterator<_Max>& it) const { return ptr != it.ptr ? !(_Max < ptr && _Max < it.ptr) : false; }
        constexpr inline bool operator==(const iterator<_Max>& it) const { return ptr != it.ptr ? (_Max < ptr && _Max < it.ptr) : true; }
    };
    constexpr PrimeListC::iterator<Max> begin() const { return PrimeListC::iterator<Max>(*this, 0); }
    constexpr PrimeListC::iterator<Max> end() const { return PrimeListC::iterator<Max>(*this, Max); }

    map<ll, int> division(ll number) const {
        map<ll, int> div;
        for (int i = 0; number > 1 && i < Max; ++i) {
            ll p = d_[i]; int c = 0;
            while (number / p * p == number) ++c, number /= p;
            if (c > 0) div[p] = c;
        }
        return div;
    }
};

//

PrimeListC<500010> P;
Combinatorics<500010, MD> C;
int N, K;

//


int main() {
    scanner >> N >> K;
    
    if (N == 1) bye("0"); // 分かりにくいです
    
    ll ans = 0;
    for (auto d1 : P) {
        if (d1*2 > N) break;
        for (auto d2 : P) {
            if (d2*2 > N) break;
            if (d1 > d2) continue;
            auto d = d1*d2;
            if (d1 == d2) d = d1;
            int n = N, k = K;
            if (n%d == 0 && k%d == 0) {
                n /= d;
                k /= d;
                ll v = C.nCr(n, k);
                if (d1 == d2)
                    ans += v;
                else
                    ans += MD-v;
                
                ans %= MD;
                // cout << d << ' ' << v << endl;
            }
        }
    }
    cout << ans << endl;
    
    return 0;
}

yukicoder

結果

テストケース

ソースコード