首页 > 其他分享 >AtCoder Regular Contest 148 E ≥ K

AtCoder Regular Contest 148 E ≥ K

时间:2023-05-29 22:57:47浏览次数:52  
标签:AtCoder typedef frac ll 148 Regular maxn long mod

洛谷传送门

AtCoder 传送门

是一道不错的计数。

考察合法序列的形态,容易发现:

  1. 不能出现两个 \(< \frac{k}{2}\) 的数相邻;
  2. 如果 \(x < \frac{k}{2}, y \ge \frac{k}{2}\),那么 \(|y - \frac{k}{2}| \ge |x - \frac{k}{2}|\)。

考虑不直接排列,而是按照一定的顺序插入。考虑按照 \(|x - \frac{k}{2}|\) 从大到小插入,并且如果 \(|x - \frac{k}{2}|\) 相同,那么就让 \(\ge \frac{k}{2}\) 的数先插入,这样能保证不会出现第二种不合法情况。因为题目不区分相同的数,于是把所有相同的数放一起考虑。

于是只用限制第一个不合法情况。维护一个可用位置数 \(s\),然后:

  • 遇到 \(x < \frac{k}{2}\),出现次数为 \(y\),那我们有 \(\binom{s}{y}\) 种方案插入,并且可用位置数减去 \(y\)(之后不能在它们两侧放数了);
  • 遇到 \(x \ge \frac{k}{2}\),出现次数为 \(y\),这个插入的方案数相当于,把这 \(y\) 个数划分成 \(s\) 段,把这 \(s\) 段依次插入,允许有空段。这个用插板法可得方案数为 \(\binom{s + y - 1}{s - 1}\),并且可用位置数加上 \(y\)。

最后由乘法原理把每一步的方案数乘起来就是最终答案。

时间复杂度 \(O(n \log n)\),瓶颈在排序。

code
// Problem: E - ≥ K
// Contest: AtCoder - AtCoder Regular Contest 148
// URL: https://atcoder.jp/contests/arc148/tasks/arc148_e
// Memory Limit: 1024 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
#define pb emplace_back
#define fst first
#define scd second
#define mems(a, x) memset((a), (x), sizeof(a))

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef double db;
typedef long double ldb;
typedef pair<ll, ll> pii;

const int maxn = 200100;
const ll mod = 998244353;

inline ll qpow(ll b, ll p) {
	ll res = 1;
	while (p) {
		if (p & 1) {
			res = res * b % mod;
		}
		b = b * b % mod;
		p >>= 1;
	}
	return res;
}

ll n, m, a[maxn], fac[maxn], ifac[maxn], tot;
pii b[maxn];

inline ll C(ll n, ll m) {
	if (n < m || n < 0 || m < 0) {
		return 0;
	} else {
		return fac[n] * ifac[m] % mod * ifac[n - m] % mod;
	}
}

void solve() {
	scanf("%lld%lld", &n, &m);
	map<ll, ll> mp;
	for (int i = 1; i <= n; ++i) {
		scanf("%lld", &a[i]);
		++mp[a[i]];
	}
	for (pii p : mp) {
		b[++tot] = p;
	}
	fac[0] = 1;
	for (int i = 1; i <= n; ++i) {
		fac[i] = fac[i - 1] * i % mod;
	}
	ifac[n] = qpow(fac[n], mod - 2);
	for (int i = n - 1; ~i; --i) {
		ifac[i] = ifac[i + 1] * (i + 1) % mod;
	}
	sort(b + 1, b + tot + 1, [&](pii a, pii b) {
		return abs(a.fst * 2 - m) > abs(b.fst * 2 - m) || (abs(a.fst * 2 - m) == abs(b.fst * 2 - m) && a.fst * 2 > m);
	});
	ll ans = 1, s = 1;
	for (int i = 1; i <= tot; ++i) {
		if (b[i].fst * 2 < m) {
			ans = ans * C(s, b[i].scd) % mod;
			s -= b[i].scd;
		} else {
			ans = ans * C(s + b[i].scd - 1, s - 1) % mod;
			s += b[i].scd;
		}
	}
	printf("%lld\n", ans);
}

int main() {
	int T = 1;
	// scanf("%d", &T);
	while (T--) {
		solve();
	}
	return 0;
}

标签:AtCoder,typedef,frac,ll,148,Regular,maxn,long,mod
From: https://www.cnblogs.com/zltzlt-blog/p/17441914.html

相关文章

  • AtCoder Beginner Contest 290(D,E)
    AtCoderBeginnerContest290(D,E)D(思维,数学)D这个题的大意就是我们需要标记\(n\)个位置,他是这样标记的,一开始有一个初始值为\(0\)的\(x\),第一个标记的是\(0\)位置,然后下一步,我们把\(x\)变成\((x+d)modn\),如果这个位置没有被标记,否则把\(x\)变成\((x+1)modn\),这个是没有......
  • CodeForces 1830C Hyperregular Bracket Strings
    洛谷传送门CF传送门每一步思路都非常自然的题。考虑先从一些简单的case入手。1.\(k=0\)设\(g_i\)为长度为\(i\)的合法括号串数。显然\(\foralli\nmid2,g_i=0\)。对于偶数的\(i\),这是一个经典问题,\(g_i\)就是第\(\frac{i}{2}\)项卡特兰数,因此\(g_i=\bi......
  • AtCoder Regular Contest 161 E Not Dyed by Majority (Cubic Graph)
    洛谷传送门AtCoder传送门给构造题提供了一种新的思路。如果答案占总方案数的比较大,可以考虑随机化。例如本题,考虑随机化,难点变成判断一个方案是否可行。考虑2-SAT,如果一个点是\(\text{B}\),那么对于这个点的边,有一条边的另一个端点是\(\text{W}\),那么其他两个都是\(\text{......
  • AtCoder Beginner Contest 303 题解 A - E
    A-SimilarString题目大意忽略0和o的差别以及1和l的差别比较两个字符串。解题思路可以硬求,直接写个超长的if判断一下。可以对两个字符串进行修改,0都换成o,1都换成l,然后直接比较两个字符串。ACCode硬求#include<iostream>#include<algorithm>#include<cstring>#i......
  • AtCoder Beginner Contest 303 G Bags Game
    洛谷传送门AtCoder传送门经典题,考虑区间dp,\(f_{l,r}\)表示只考虑\([l,r]\)区间,先手得分减后手得分最大值。对于第一种操作直接\(f_{l,r}\gets\max(a_l-f_{l+1,r},a_r-f_{l,r-1})\),第二种操作,考虑枚举\([l,r]\)长为\(r-l+1-B\)的子段,即可转移。第三种操......
  • AtCoder Beginner Contest 298(D,F)
    AtCoderBeginnerContest298(D,F)D(思维,模拟,快速幂)D大意是最初有一个数字\(1\),然后进行\(q\)个操作有三种操作\(1\),输入\(1,x\),在原来的数字后面添加一个尾数,例如原本的数是\(12\),输入了\(15\),数字变成了\(125\)\(2\),输入\(2\),把原来的数字第一位数删除,例如原本的数是......
  • AtCoder Beginner Contest 299(E,F)
    AtCoderBeginnerContest299(E,F)E(最短路)E题目大意为有\(n\)个点和\(m\)条边,我们我个这些点匹配颜色(有两种颜色),但是要满足下面的条件必须由一个点的颜色是\(1\)然后给出\(k\)点限制对于\(p_i\)这一个点,离他最近的一个颜色为\(1\)的点的最近距离为\(d_i\)既然知道某个点......
  • Atcoder Grand Contest 062 D - Walk Around Neighborhood
    csy/bxwjz/bx首先将\(a\)排序,如果\(\sum\limits_{i=1}^{n-1}a_i<a_n\)显然就是\(-1\),否则必然有解。先发现一个trivial的事情,就是答案一定位于\([\dfrac{a_n}{2},a_n]\)中,这是因为我们判掉无解的条件之后,我们必然可以用前面的步数走到以\((a_n,0),(0,a_n),(-a_n,0),(......
  • 1480. Running Sum of 1d Array刷题笔记
    简单难度的题classSolution:defrunningSum(self,nums:List[int])->List[int]:foriinrange(1,len(nums)):nums[i]+=nums[i-1]returnnums......
  • AtCoder Regular Contest 139 E Wazir
    洛谷传送门AtCoder传送门好题。这种题一般可以考虑,观察最优解的性质,对于性质计数。发现如果\(n,m\)均为偶数,可以放满。就是类似这样:#.#.#..#.#.##.#.#..#.#.#因此答案就是\(2\)。如果\(n,m\)有一个为偶数,不妨假设\(n\)为偶数。那么最优解形似:#.#...#..##..#......