結果

問題 No.1318 ABCD quadruplets
ユーザー PachicobuePachicobue
提出日時 2020-12-15 01:57:43
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 352 ms / 2,000 ms
コード長 6,621 bytes
コンパイル時間 2,530 ms
コンパイル使用メモリ 207,228 KB
実行使用メモリ 4,372 KB
最終ジャッジ日時 2023-10-20 05:35:51
合計ジャッジ時間 6,108 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 2 ms
4,348 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 2 ms
4,348 KB
testcase_06 AC 2 ms
4,348 KB
testcase_07 AC 2 ms
4,348 KB
testcase_08 AC 1 ms
4,348 KB
testcase_09 AC 2 ms
4,348 KB
testcase_10 AC 2 ms
4,348 KB
testcase_11 AC 2 ms
4,348 KB
testcase_12 AC 2 ms
4,348 KB
testcase_13 AC 7 ms
4,348 KB
testcase_14 AC 133 ms
4,348 KB
testcase_15 AC 4 ms
4,348 KB
testcase_16 AC 171 ms
4,348 KB
testcase_17 AC 10 ms
4,348 KB
testcase_18 AC 115 ms
4,348 KB
testcase_19 AC 45 ms
4,348 KB
testcase_20 AC 7 ms
4,348 KB
testcase_21 AC 20 ms
4,348 KB
testcase_22 AC 10 ms
4,348 KB
testcase_23 AC 347 ms
4,348 KB
testcase_24 AC 317 ms
4,372 KB
testcase_25 AC 64 ms
4,372 KB
testcase_26 AC 34 ms
4,372 KB
testcase_27 AC 232 ms
4,372 KB
testcase_28 AC 78 ms
4,372 KB
testcase_29 AC 99 ms
4,372 KB
testcase_30 AC 351 ms
4,372 KB
testcase_31 AC 70 ms
4,372 KB
testcase_32 AC 352 ms
4,372 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using ll = long long;
using uint = unsigned int;
using ull = unsigned long long;
using ld = long double;
template<typename T> using max_heap = std::priority_queue<T>;
template<typename T> using min_heap = std::priority_queue<T, std::vector<T>, std::greater<T>>;
constexpr int popcount(const ull v) { return v ? __builtin_popcountll(v) : 0; }
constexpr int log2p1(const ull v) { return v ? 64 - __builtin_clzll(v) : 0; }
constexpr int lsbp1(const ull v) { return __builtin_ffsll(v); }
constexpr int clog(const ull v) { return v ? log2p1(v - 1) : 0; }
constexpr ull ceil2(const ull v) { return 1ULL << clog(v); }
constexpr ull floor2(const ull v) { return v ? (1ULL << (log2p1(v) - 1)) : 0ULL; }
constexpr bool btest(const ull mask, const int ind) { return (mask >> ind) & 1ULL; }
template<typename T> void bset(T& mask, const int ind) { mask |= ((T)1 << ind); }
template<typename T> void breset(T& mask, const int ind) { mask &= ~((T)1 << ind); }
template<typename T> void bflip(T& mask, const int ind) { mask ^= ((T)1 << ind); }
template<typename T> void bset(T& mask, const int ind, const bool b) { (b ? bset(mask, ind) : breset(mask, ind)); }
template<typename T> bool chmin(T& a, const T& b) { return (a > b ? a = b, true : false); }
template<typename T> bool chmax(T& a, const T& b) { return (a < b ? a = b, true : false); }
template<typename T> constexpr T inf_v = std::numeric_limits<T>::max() / 4;
template<typename Real> constexpr Real pi_v = Real{3.141592653589793238462643383279502884};
template<typename T> constexpr T TEN(const int n) { return n == 0 ? T{1} : TEN<T>(n - 1) * T{10}; }
template<typename F> struct fix : F
{
    fix(F&& f) : F{std::forward<F>(f)} {}
    template<typename... Args> auto operator()(Args&&... args) const { return F::operator()(*this, std::forward<Args>(args)...); }
};
template<typename T, int n, int i = 0>
auto nd_array(int const (&szs)[n], const T x = T{})
{
    if constexpr (i == n) {
        return x;
    } else {
        return std::vector(szs[i], nd_array<T, n, i + 1>(szs, x));
    }
}
class printer
{
public:
    printer(std::ostream& os_ = std::cout) : m_os{os_} { m_os << std::fixed << std::setprecision(15); }
    template<typename... Args> int ln(const Args&... args) { return dump(args...), m_os << '\n', 0; }
    template<typename... Args> int el(const Args&... args) { return dump(args...), m_os << std::endl, 0; }
private:
    template<typename T> void dump(const T& v) { m_os << v; }
    template<typename T> void dump(const std::vector<T>& vs)
    {
        for (int i = 0; i < (int)vs.size(); i++) { m_os << (i ? " " : ""), dump(vs[i]); }
    }
    template<typename T> void dump(const std::vector<std::vector<T>>& vss)
    {
        for (int i = 0; i < (int)vss.size(); i++) { m_os << (0 <= i or i + 1 < (int)vss.size() ? "\n" : ""), dump(vss[i]); }
    }
    template<typename T, typename... Args> int dump(const T& v, const Args&... args) { return dump(v), m_os << ' ', dump(args...), 0; }
    std::ostream& m_os;
};
printer out;
class scanner
{
public:
    scanner(std::istream& is_ = std::cin) : m_is{is_} { m_is.tie(nullptr), std::ios::sync_with_stdio(false); }
    template<typename T> T val()
    {
        T v;
        return m_is >> v, v;
    }
    template<typename T> T val(const T offset) { return val<T>() - offset; }
    template<typename T> std::vector<T> vec(const int n)
    {
        return make_v<T>(n, [this]() { return val<T>(); });
    }
    template<typename T> std::vector<T> vec(const int n, const T offset)
    {
        return make_v<T>(n, [this, offset]() { return val<T>(offset); });
    }
    template<typename T> std::vector<std::vector<T>> vvec(const int n0, const int n1)
    {
        return make_v<std::vector<T>>(n0, [this, n1]() { return vec<T>(n1); });
    }
    template<typename T> std::vector<std::vector<T>> vvec(const int n0, const int n1, const T offset)
    {
        return make_v<std::vector<T>>(n0, [this, n1, offset]() { return vec<T>(n1, offset); });
    }
    template<typename... Args> auto tup() { return std::tuple<std::decay_t<Args>...>{val<Args>()...}; }
    template<typename... Args> auto tup(const Args&... offsets) { return std::tuple<std::decay_t<Args>...>{val<Args>(offsets)...}; }
private:
    template<typename T, typename F>
    std::vector<T> make_v(const int n, F f)
    {
        std::vector<T> ans;
        for (int i = 0; i < n; i++) { ans.push_back(f()); }
        return ans;
    }
    std::istream& m_is;
};
scanner in;
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
int main()
{
    const auto [N, M] = in.tup<int, int>();
    std::vector<int> vs(4, 0);
    std::vector<int> ss(4, 0), ts(4, 0);
    std::vector<int> ans(N + 1, 0);
    for (vs[0] = 0; vs[0] <= M; vs[0]++) {
        ss[0] = vs[0], ts[0] = vs[0] * vs[0];
        for (vs[1] = vs[0]; vs[1] <= M; vs[1]++) {
            ss[1] = ss[0] + vs[1], ts[1] = ts[0] + ss[1] * vs[1];
            if (ts[1] > N) { break; }
            for (vs[2] = vs[1]; vs[2] <= M; vs[2]++) {
                ss[2] = ss[1] + vs[2], ts[2] = ts[1] + ss[2] * vs[2];
                if (ts[2] > N) { break; }
                for (vs[3] = vs[2]; vs[3] <= M; vs[3]++) {
                    ss[3] = ss[2] + vs[3], ts[3] = ts[2] + ss[3] * vs[3];
                    if (ts[3] > N) { break; }
                    int way = 24;
                    if (vs[0] == vs[1]) {
                        if (vs[1] == vs[2]) {
                            if (vs[2] == vs[3]) {
                                way = 1;
                            } else {
                                way = 4;
                            }
                        } else {
                            if (vs[2] == vs[3]) {
                                way = 6;
                            } else {
                                way = 12;
                            }
                        }
                    } else {
                        if (vs[1] == vs[2]) {
                            if (vs[2] == vs[3]) {
                                way = 4;
                            } else {
                                way = 12;
                            }
                        } else {
                            if (vs[2] == vs[3]) {
                                way = 12;
                            } else {
                                way = 24;
                            }
                        }
                    }
                    ans[ts[3]] += way;
                }
            }
        }
    }
    for (int i = 0; i <= N; i++) { out.ln(ans[i]); }
    return 0;
}
0