Submission #1679966


Source Code Expand

// need
#include <iostream>
#include <algorithm>

// data structure
#include <bitset>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <utility>
#include <vector>
#include <array>
//#include <unordered_map>
#include <complex>
//#include <deque>
#include <valarray>

// stream
//#include <istream>
//#include <sstream>
//#include <ostream>
//#include <fstream>

// etc
#include <cassert>
#include <cmath>
#include <functional>
#include <iomanip>
#include <chrono>
#include <random>
#include <numeric>

// input
#define INIT std::ios::sync_with_stdio(false);std::cin.tie(0);
#define VAR(type, ...)type __VA_ARGS__;MACRO_VAR_Scan(__VA_ARGS__);
template<typename T> void MACRO_VAR_Scan(T& t) { std::cin >> t; }
template<typename First, typename...Rest>void MACRO_VAR_Scan(First& first, Rest&...rest) { std::cin >> first; MACRO_VAR_Scan(rest...); }
#define VEC_ROW(type, n, ...)std::vector<type> __VA_ARGS__;MACRO_VEC_ROW_Init(n, __VA_ARGS__); for(int i=0; i<n; ++i){MACRO_VEC_ROW_Scan(i, __VA_ARGS__);}
template<typename T> void MACRO_VEC_ROW_Init(int n, T& t) { t.resize(n); }
template<typename First, typename...Rest>void MACRO_VEC_ROW_Init(int n, First& first, Rest&...rest) { first.resize(n); MACRO_VEC_ROW_Init(n, rest...); }
template<typename T> void MACRO_VEC_ROW_Scan(int p, T& t) { std::cin >> t[p]; }
template<typename First, typename...Rest>void MACRO_VEC_ROW_Scan(int p, First& first, Rest&...rest) { std::cin >> first[p]; MACRO_VEC_ROW_Scan(p, rest...); }
#define VEC(type, c, n) std::vector<type> c(n);for(auto& i:c)std::cin>>i;
#define MAT(type, c, m, n) std::vector<std::vector<type>> c(m, std::vector<type>(n));for(auto& r:c)for(auto& i:r)std::cin>>i;

// output
#define OUT(d) std::cout<<d;
#define FOUT(n, d) std::cout<<std::fixed<<std::setprecision(n)<<d;
#define SOUT(n, c, d) std::cout<<std::setw(n)<<std::setfill(c)<<d;
#define SP std::cout<<" ";
#define TAB std::cout<<"\t";
#define BR std::cout<<"\n";
#define SPBR(i, n) std::cout<<(i + 1 == n ? '\n' : ' ');
#define ENDL std::cout<<std::endl;
#define FLUSH std::cout<<std::flush;
#define SHOW(d) {std::cerr << #d << "\t:" << d << "\n";}
#define SHOWVECTOR(v) {std::cerr << #v << "\t:";for(const auto& xxx : v){std::cerr << xxx << " ";}std::cerr << "\n";}
#define SHOWVECTOR2(v) {std::cerr << #v << "\t:\n";for(const auto& xxx : v){for(const auto& yyy : xxx){std::cerr << yyy << " ";}std::cerr << "\n";}}
#define SHOWQUEUE(a) {auto tmp(a);std::cerr << #a << "\t:";while(!tmp.empty()){std::cerr << tmp.front() << " ";tmp.pop();}std::cerr << "\n";}

// utility
#define ALL(a) (a).begin(),(a).end()
#define FOR(i, a, b) for(int i=(a);i<(b);++i)
#define RFOR(i, a, b) for(int i=(b)-1;i>=(a);--i)
#define REP(i, n) for(int i=0;i<int(n);++i)
#define RREP(i, n) for(int i=int(n)-1;i>=0;--i)
#define FORLL(i, a, b) for(ll i=ll(a);i<ll(b);++i)
#define RFORLL(i, a, b) for(ll i=ll(b)-1;i>=ll(a);--i)
#define REPLL(i, n) for(ll i=0;i<ll(n);++i)
#define RREPLL(i, n) for(ll i=ll(n)-1;i>=0;--i)
#define IN(a, x, b) (a<=x && x<b)
template<typename T> inline T CHMAX(T& a, const T b) { return a = (a < b) ? b : a; }
template<typename T> inline T CHMIN(T& a, const T b) { return a = (a > b) ? b : a; }
#define EXCEPTION(msg) throw std::string("Exception : " msg " [ in ") + __func__ + " : " + std::to_string(__LINE__) + " lines ]"
#define TRY(cond, msg) try {if (cond) EXCEPTION(msg);}catch (std::string s) {std::cerr << s << std::endl;}
void CHECKTIME(std::function<void()> f) { auto start = std::chrono::system_clock::now(); f(); auto end = std::chrono::system_clock::now(); auto res = std::chrono::duration_cast<std::chrono::nanoseconds>((end - start)).count(); std::cerr << "[Time:" << res << "ns  (" << res / (1.0e9) << "s)]\n"; }

// test
template<class T> std::vector<std::vector<T>> VV(int n) {
	return std::vector<std::vector<T>>(n);
}
template<class T> std::vector<std::vector<T>> VV(int n, int m, T init = T()) {
	return std::vector<std::vector<T>>(n, std::vector<T>(m, init));
}
template<typename S, typename T>
std::ostream& operator<<(std::ostream& os, std::pair<S, T> p) {
	os << "(" << p.first << ", " << p.second << ")"; return os;
}

// type/const

#define int ll
using ll = long long;
using ull = unsigned long long;
using ld = long double;
using PAIR = std::pair<int, int>;
using PAIRLL = std::pair<ll, ll>;
constexpr int INFINT = 1 << 30;                          // 1.07x10^ 9
constexpr int INFINT_LIM = (1LL << 31) - 1;              // 2.15x10^ 9
constexpr ll INFLL = 1LL << 60;                          // 1.15x10^18
constexpr ll INFLL_LIM = (1LL << 62) - 1 + (1LL << 62);  // 9.22x10^18
constexpr double EPS = 1e-7;
constexpr int MOD = 1000000007;
constexpr double PI = 3.141592653589793238462643383279;

ll powMod(ll n, ll p, ll mod) {
	ll res = 1;
	while (p) {
		if (p & 1) res *= n, res %= mod;
		n *= n; n %= mod;
		p >>= 1;
	}
	return res;
}
ll invMod(ll n, ll mod) {
	return powMod(n, MOD - 2, MOD);
}

const signed FACT_MAX_N = 1000006;
signed fact[FACT_MAX_N];
signed factInv[FACT_MAX_N];
struct INIT_FACT {
	INIT_FACT() {
		fact[0] = 1;
		for (int i = 1; i < FACT_MAX_N; ++i) fact[i] = (long long)i * fact[i - 1] % MOD;
		factInv[FACT_MAX_N - 1] = powMod(fact[FACT_MAX_N - 1], MOD - 2, MOD);
		for (int i = FACT_MAX_N - 2; i >= 0; --i) {
			factInv[i] = (long long)(i + 1) * factInv[i + 1] % MOD;
		}
	}
} init_fact;

signed main() {
	INIT;
	VAR(int, n, x, y);
	VEC_ROW(int, n, c, w);
	std::map<int, std::vector<int>> map;
	REP(i, n) {
		map[c[i]].emplace_back(w[i]);
	}
	std::vector<PAIR> pv;
	for (auto& v : map) {
		std::sort(ALL(v.second));
		pv.emplace_back(PAIR(v.second[0], v.first));
	}
	int m = pv.size();
	std::sort(ALL(pv));
	std::vector<std::vector<int>> a(m);
	std::vector<int> ccnt(m, 0);
	std::map<int, std::map<int, bool>> checked;
	REP(i, m) a[i] = map[pv[i].second];
	checked[0][0];
	int ans = 1;
	FOR(i, 1, m) {
		if (a[0][0] + a[i][0] > y) break;
		REP(j, a[i].size()) {
			if (a[0][0] + a[i][j] <= y || a[i][0] + a[i][j] <= x) {
				++ccnt[i];
				checked[i][j];
			}
		}
		(ans *= factInv[ccnt[i]]) %= MOD;
	}
	if (a.size() >= 2 && a[0][0] + a[1][0] <= y) {
		REP(j, a[0].size()) {
			if (a[1][0] + a[0][j] <= y || a[0][0] + a[0][j] <= x) {
				++ccnt[0];
				checked[0][j];
			}
		}
		(ans *= factInv[ccnt[0]]) %= MOD;
	}
	int cnt = 0;
	for (const auto& m1 : checked) {
		for (const auto& m2 : m1.second) ++cnt;
	}
	(ans *= fact[cnt]) %= MOD;
	OUT(ans)BR;
	return 0;
}

Submission Info

Submission Time
Task D - Colorful Balls
User satanic0258
Language C++14 (GCC 5.4.1)
Score 1000
Code Size 6603 Byte
Status AC
Exec Time 399 ms
Memory 59244 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 1000 / 1000
Status
AC × 3
AC × 57
Set Name Test Cases
Sample 00_example_01.txt, 00_example_02.txt, 00_example_03.txt
All 00_example_01.txt, 00_example_02.txt, 00_example_03.txt, 01.txt, 02.txt, 03.txt, 04.txt, 05.txt, 06.txt, 07.txt, 08.txt, 09.txt, 10.txt, 11.txt, 12.txt, 13.txt, 14.txt, 15.txt, 16.txt, 17.txt, 18.txt, 19.txt, 20.txt, 21.txt, 22.txt, 23.txt, 24.txt, 25.txt, 26.txt, 27.txt, 28.txt, 29.txt, 30.txt, 31.txt, 32.txt, 33.txt, 34.txt, 35.txt, 36.txt, 37.txt, 38.txt, 39.txt, 40.txt, 41.txt, 42.txt, 43.txt, 44.txt, 45.txt, 46.txt, 47.txt, 48.txt, 49.txt, 50.txt, 51.txt, 52.txt, 53.txt, 54.txt
Case Name Status Exec Time Memory
00_example_01.txt AC 15 ms 8064 KB
00_example_02.txt AC 15 ms 8064 KB
00_example_03.txt AC 15 ms 8064 KB
01.txt AC 15 ms 8192 KB
02.txt AC 15 ms 8064 KB
03.txt AC 20 ms 9472 KB
04.txt AC 15 ms 8064 KB
05.txt AC 17 ms 8704 KB
06.txt AC 15 ms 8192 KB
07.txt AC 100 ms 23928 KB
08.txt AC 15 ms 8192 KB
09.txt AC 39 ms 13824 KB
10.txt AC 33 ms 12672 KB
11.txt AC 16 ms 8192 KB
12.txt AC 15 ms 8064 KB
13.txt AC 21 ms 9984 KB
14.txt AC 15 ms 8064 KB
15.txt AC 57 ms 18296 KB
16.txt AC 98 ms 20468 KB
17.txt AC 24 ms 10240 KB
18.txt AC 35 ms 12160 KB
19.txt AC 58 ms 16384 KB
20.txt AC 315 ms 48624 KB
21.txt AC 299 ms 47600 KB
22.txt AC 237 ms 40048 KB
23.txt AC 367 ms 56048 KB
24.txt AC 345 ms 39920 KB
25.txt AC 257 ms 48240 KB
26.txt AC 399 ms 57072 KB
27.txt AC 226 ms 35184 KB
28.txt AC 383 ms 57200 KB
29.txt AC 372 ms 53104 KB
30.txt AC 278 ms 37360 KB
31.txt AC 366 ms 57328 KB
32.txt AC 354 ms 41584 KB
33.txt AC 313 ms 56560 KB
34.txt AC 298 ms 51444 KB
35.txt AC 89 ms 24704 KB
36.txt AC 129 ms 29948 KB
37.txt AC 90 ms 25088 KB
38.txt AC 89 ms 24320 KB
39.txt AC 94 ms 25088 KB
40.txt AC 330 ms 57584 KB
41.txt AC 361 ms 52204 KB
42.txt AC 326 ms 50280 KB
43.txt AC 273 ms 42220 KB
44.txt AC 362 ms 59244 KB
45.txt AC 286 ms 43116 KB
46.txt AC 343 ms 51692 KB
47.txt AC 63 ms 14448 KB
48.txt AC 64 ms 14448 KB
49.txt AC 64 ms 14448 KB
50.txt AC 205 ms 44144 KB
51.txt AC 182 ms 37744 KB
52.txt AC 106 ms 22132 KB
53.txt AC 107 ms 22132 KB
54.txt AC 109 ms 22132 KB