제출 #760168

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
760168		drdilyor	Party (INOI20_party)	C++17	0 / 100	141 ms	296 KiB

Main

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
#define DEBUG
#if defined(ONPC) && !defined(NODEBUG)
#define debug(args...) cout << "["<< #args << "]= "; debug_out(args);
#else
#define debug(args...) 42;
#endif
void debug_out() {
    cout << endl;
}
template<typename H, typename... T>
void debug_out(H h, T... t) {
    cout << h;
    if (sizeof...(t)) cout << ", ";
    debug_out(t...);
}

constexpr int mod = 1e9 + 7;
ll modpow(ll a, ll b) {
    a %= mod;
    ll res = 1;
    while (b) {
        if (b&1) res = res * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return res;
}
ll inv(ll x) {
    return modpow(x, mod-2);
}
ll pw2(ll x) {
    return x < 0 ? 0 :(1LL << x);
}
ll sub(ll a, ll b) {
    a -= b;
    return a < 0 ? a + mod : a;
}

ll solve(ll n) {
    ll k = __lg(n+1);
    if ((1ll<<k)-1 != n) return -1;
    debug(n, k);
    auto comb = [&](int chosen)->ll{ return sub(modpow(2, n - chosen), 1); };
    ll res = 0;

    for (int i = 0; i < k; i++) {
        for (int d = 1; d <= k*2; d++) {
            ll cnt = 0, perim=0;
            int lastj = 0;
            for (int j = i; j >= 0 && i - j < d; j--) {
                lastj = j;
                ll size = i == j ? pw2(k-j)-1 : pw2(k - j -1);
                int depth = min((int)k, d - (i - j));
                ll need = i == j ? pw2(depth) - 1 : (pw2(depth)-1) - (pw2(depth-1)-1);
                debug(i, d, depth, j, need, size, perim);
                ll cperim = need >= size ? 0 : i == j ? need+1 : need;
                perim += cperim;
                cnt += min(size, need);
                if (cperim) assert(need + cperim <= size);
                assert(perim + cnt <= n);
            }
            if (lastj) perim++;
            if (n == cnt) break;
            ll level = pw2(i);
            ll p = comb(cnt);
            if (cnt + perim < n)
                p = sub(p, comb(cnt + perim)); 
            debug(i, d, cnt, level, perim, p);
            res += (d%mod) * (level%mod) % mod * (p%mod);
            res %= mod;
            debug(res);
        }
    }
    debug(res, comb(0));
    return res * inv(comb(0)) % mod;
}

int main() {
    cin.tie(0)->sync_with_stdio(0);

    int q; cin >> q;
    while (q--) {
        ll n; cin >> n;
        cout << solve(n) << '\n';
    }
}

컴파일 시 표준 에러 (stderr) 메시지

Main.cpp: In function 'll solve(ll)':
Main.cpp:8:24: warning: statement has no effect [-Wunused-value]
    8 | #define debug(args...) 42;
      |                        ^~
Main.cpp:45:5: note: in expansion of macro 'debug'
   45 |     debug(n, k);
      |     ^~~~~
Main.cpp:8:24: warning: statement has no effect [-Wunused-value]
    8 | #define debug(args...) 42;
      |                        ^~
Main.cpp:58:17: note: in expansion of macro 'debug'
   58 |                 debug(i, d, depth, j, need, size, perim);
      |                 ^~~~~
Main.cpp:8:24: warning: statement has no effect [-Wunused-value]
    8 | #define debug(args...) 42;
      |                        ^~
Main.cpp:71:13: note: in expansion of macro 'debug'
   71 |             debug(i, d, cnt, level, perim, p);
      |             ^~~~~
Main.cpp:8:24: warning: statement has no effect [-Wunused-value]
    8 | #define debug(args...) 42;
      |                        ^~
Main.cpp:74:13: note: in expansion of macro 'debug'
   74 |             debug(res);
      |             ^~~~~
Main.cpp:8:24: warning: statement has no effect [-Wunused-value]
    8 | #define debug(args...) 42;
      |                        ^~
Main.cpp:77:5: note: in expansion of macro 'debug'
   77 |     debug(res, comb(0));
      |     ^~~~~

• Subtask 1: 0 / 7
• Subtask 2: 0 / 23
• Subtask 3: 0 / 33
• Subtask 4: 0 / 37