#393553 (C++17) No.916 Encounter On A Tree

提出ソース

結果

問題	No.916 Encounter On A Tree
ユーザー	はまやんはまやん
提出日時	2019-10-25 22:56:18
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	29 ms / 2,000 ms
コード長	5,061 bytes
コンパイル時間	2,183 ms
コンパイル使用メモリ	208,300 KB
実行使用メモリ	19,128 KB
最終ジャッジ日時	2024-09-13 06:38:54
合計ジャッジ時間	5,307 ms
ジャッジサーバーID （参考情報）	judge4 / judge1

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	26 ms 18,768 KB
testcase_01	AC	26 ms 18,844 KB
testcase_02	AC	27 ms 18,876 KB
testcase_03	AC	27 ms 18,884 KB
testcase_04	AC	26 ms 18,924 KB
testcase_05	AC	26 ms 18,936 KB
testcase_06	AC	27 ms 18,944 KB
testcase_07	AC	27 ms 18,916 KB
testcase_08	AC	27 ms 18,936 KB
testcase_09	AC	29 ms 18,888 KB
testcase_10	AC	28 ms 18,952 KB
testcase_11	AC	28 ms 18,940 KB
testcase_12	AC	27 ms 18,772 KB
testcase_13	AC	27 ms 18,856 KB
testcase_14	AC	28 ms 18,764 KB
testcase_15	AC	27 ms 18,880 KB
testcase_16	AC	27 ms 18,832 KB
testcase_17	AC	29 ms 18,792 KB
testcase_18	AC	27 ms 18,860 KB
testcase_19	AC	27 ms 18,936 KB
testcase_20	AC	25 ms 18,876 KB
testcase_21	AC	26 ms 18,876 KB
testcase_22	AC	27 ms 18,796 KB
testcase_23	AC	26 ms 18,796 KB
testcase_24	AC	27 ms 18,916 KB
testcase_25	AC	27 ms 18,888 KB
testcase_26	AC	28 ms 18,876 KB
testcase_27	AC	25 ms 18,912 KB
testcase_28	AC	27 ms 18,916 KB
testcase_29	AC	28 ms 19,128 KB
testcase_30	AC	27 ms 18,796 KB
testcase_31	AC	27 ms 18,932 KB
testcase_32	AC	27 ms 18,920 KB
testcase_33	AC	27 ms 18,828 KB
testcase_34	AC	28 ms 18,880 KB
testcase_35	AC	25 ms 18,892 KB
testcase_36	AC	27 ms 18,768 KB
testcase_37	AC	26 ms 18,956 KB
testcase_38	AC	27 ms 18,928 KB
testcase_39	AC	26 ms 18,796 KB
testcase_40	AC	27 ms 18,868 KB
testcase_41	AC	26 ms 18,916 KB
testcase_42	AC	27 ms 18,944 KB
testcase_43	AC	27 ms 18,920 KB
testcase_44	AC	26 ms 18,832 KB
testcase_45	AC	27 ms 18,980 KB
testcase_46	AC	27 ms 18,832 KB
testcase_47	AC	26 ms 18,880 KB
testcase_48	AC	27 ms 18,780 KB
testcase_49	AC	27 ms 18,952 KB
testcase_50	AC	28 ms 18,768 KB
testcase_51	AC	28 ms 18,796 KB
testcase_52	AC	27 ms 18,892 KB
testcase_53	AC	27 ms 18,952 KB
testcase_54	AC	27 ms 18,888 KB
testcase_55	AC	28 ms 18,872 KB
testcase_56	AC	27 ms 18,872 KB
testcase_57	AC	28 ms 18,832 KB
testcase_58	AC	27 ms 18,976 KB
testcase_59	AC	27 ms 18,868 KB
testcase_60	AC	26 ms 18,768 KB

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

ソースコード

raw source code

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<b;i++)
#define rrep(i,a,b) for(int i=a;i>=b;i--)
#define fore(i,a) for(auto &i:a)
#define all(x) (x).begin(),(x).end()
//#pragma GCC optimize ("-O3")
using namespace std; void _main(); int main() { cin.tie(0); ios::sync_with_stdio(false); _main(); }
typedef long long ll; const int inf = INT_MAX / 2; const ll infl = 1LL << 60;
template<class T>bool chmax(T& a, const T& b) { if (a < b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T& a, const T& b) { if (b < a) { a = b; return 1; } return 0; }
//---------------------------------------------------------------------------------------------------
template<int MOD> struct ModInt {
    static const int Mod = MOD; unsigned x; ModInt() : x(0) { }
    ModInt(signed sig) { x = sig < 0 ? sig % MOD + MOD : sig % MOD; }
    ModInt(signed long long sig) { x = sig < 0 ? sig % MOD + MOD : sig % MOD; }
    int get() const { return (int)x; }
    ModInt &operator+=(ModInt that) { if ((x += that.x) >= MOD) x -= MOD; return *this; }
    ModInt &operator-=(ModInt that) { if ((x += MOD - that.x) >= MOD) x -= MOD; return *this; }
    ModInt &operator*=(ModInt that) { x = (unsigned long long)x * that.x % MOD; return *this; }
    ModInt &operator/=(ModInt that) { return *this *= that.inverse(); }
    ModInt operator+(ModInt that) const { return ModInt(*this) += that; }
    ModInt operator-(ModInt that) const { return ModInt(*this) -= that; }
    ModInt operator*(ModInt that) const { return ModInt(*this) *= that; }
    ModInt operator/(ModInt that) const { return ModInt(*this) /= that; }
    ModInt inverse() const { long long a = x, b = MOD, u = 1, v = 0;
        while (b) { long long t = a / b; a -= t * b; std::swap(a, b); u -= t * v; std::swap(u, v); }
        return ModInt(u); }
    bool operator==(ModInt that) const { return x == that.x; }
    bool operator!=(ModInt that) const { return x != that.x; }
    ModInt operator-() const { ModInt t; t.x = x == 0 ? 0 : Mod - x; return t; }
};
template<int MOD> ostream& operator<<(ostream& st, const ModInt<MOD> a) { st << a.get(); return st; };
template<int MOD> ModInt<MOD> operator^(ModInt<MOD> a, unsigned long long k) {
    ModInt<MOD> r = 1; while (k) { if (k & 1) r *= a; a *= a; k >>= 1; } return r; }
template<typename T, int FAC_MAX> struct Comb { vector<T> fac, ifac;
    Comb(){fac.resize(FAC_MAX,1);ifac.resize(FAC_MAX,1);rep(i,1,FAC_MAX)fac[i]=fac[i-1]*i;
        ifac[FAC_MAX-1]=T(1)/fac[FAC_MAX-1];rrep(i,FAC_MAX-2,1)ifac[i]=ifac[i+1]*T(i+1);}
    T aPb(int a, int b) { if (b < 0 || a < b) return T(0); return fac[a] * ifac[a - b]; }
    T aCb(int a, int b) { if (b < 0 || a < b) return T(0); return fac[a] * ifac[a - b] * ifac[b]; }
    T nHk(int n, int k) { if (n == 0 && k == 0) return T(1); if (n <= 0 || k < 0) return 0;
        return aCb(n + k - 1, k); } // nHk = (n+k-1)Ck : n is separator
    T pairCombination(int n) {if(n%2==1)return T(0);return fac[n]*ifac[n/2]/(T(2)^(n/2));}
    // combination of paris for n
}; 
typedef ModInt<1000000007> mint;
/*---------------------------------------------------------------------------------------------------
　　　　　　　　　　　 ∧＿∧
　　　　　 ∧＿∧ 　（´<_｀ ）　 Welcome to My Coding Space!
　　　　 （ ´_ゝ`）　/　 ⌒i     @hamayanhamayan
　　　　／　　　＼　 　  |　|
　　　 /　　 /￣￣￣￣/　　|
　 ＿_(__ﾆつ/　    ＿/ .| .|＿＿＿＿
　 　　　＼/＿＿＿＿/　（u　⊃
---------------------------------------------------------------------------------------------------*/













int D, L, R, K;
Comb<mint, 2010101> com;
int lft[21], rht[21];
int cnt[21];
//---------------------------------------------------------------------------------------------------
void _main() {
	cin >> D >> L >> R >> K;

	lft[1] = rht[1] = 1;
	rep(d, 1, D) {
		lft[d + 1] = lft[d] * 2;
		rht[d + 1] = lft[d + 1] * 2 - 1;
	}
	rep(d, 1, D + 1) cnt[d] = rht[d] - lft[d] + 1;

	mint ans = 0;
	rep(dl, 1, D + 1) rep(dr, 1, D + 1) rep(dlca, 1, D + 1) {
		if (dlca > dl) continue;
		if (dlca > dr) continue;
		if ((dl - dlca) + (dr - dlca) != K) continue;

		if (lft[dl] <= L and L <= rht[dl] and lft[dr] <= R and R <= rht[dr]) {
			mint delta = 1;

			if (dl == dr) {
				delta *= cnt[dlca];
				delta *= cnt[dl] / cnt[dlca];
				delta *= cnt[dl] / cnt[dlca];
				delta /= 2;
			}
			else if (dl == dlca) {
				delta *= cnt[dlca];
				delta *= cnt[dr] / cnt[dlca];
			}
			else if (dr == dlca) {
				delta *= cnt[dlca];
				delta *= cnt[dl] / cnt[dlca];
			}
			else {
				delta *= cnt[dlca];
				delta *= cnt[dl] / cnt[dlca];
				delta *= cnt[dr] / cnt[dlca];
				delta /= 2;
			}

			rep(d, 1, D + 1) {
				if (dl == d and dr == d) delta *= com.fac[cnt[d] - 2];
				else if (dl == d or dr == d) delta *= com.fac[cnt[d] - 1];
				else delta *= com.fac[cnt[d]];
			}

			ans += delta;
		}
	}
	cout << ans << endl;
}

yukicoder

結果

テストケース

ソースコード