Panasonic Programming Contest 2025（AtCoder Beginner Contest 406）D-E 题解-CSDN博客

本文链接：https://blog.csdn.net/AKDreamer_HeXY/article/details/148190450

D. Garbage Removal

题意

有一个 $\times W$ 的网格， $(i, j)$ 表示第 $i$ 行第 $j$ 列的格子。在这个网格里有 $N$ 个垃圾，第 $i$ 个垃圾在 $X_i,Y_i)$ 。

有 $Q$ 次查询，分为以下两种类型：

1 x 输出第 $x$ 行垃圾的总数，并移除这行所有垃圾；
2 y 输出第 $y$ 列垃圾的总数，并移除这列所有垃圾。

思路

由于删除一行垃圾时，对应列的垃圾数也会减少。所以考虑用 $\text{set}$ 维护每行、每列的状态。

定义 set<int> X[]（大小为 $H$ ）， $X_i$ 表示第 $i$ 行有垃圾的所有列号的集合；和 set<int> Y[]（大小为 $W$ ）， $Y_i$ 表示第 $i$ 列有垃圾的所有行号的集合。

初始化将所有垃圾都加入对应集合，注意 $X$ 和 $Y$ 都要加。

若为操作 $1$ ，输出集合 $X_x$ 的大小，并且，对于其中每一个数 $i$ （列号），删除对应集合 $Y_i$ 中的 $x$ 元素，并清空集合 $X_x$ ；

若为操作 $2$ ，输出集合 $Y_y$ 的大小，并且，对于其中每一个数 $i$ （行号），删除对应集合 $X_i$ 中的 $y$ 元素，并清空集合 $Y_y$ 。

时间复杂度： $O (Q + N)$

C++ 代码

#include<bits/stdc++.h>
#define sz(v) (int)v.size()
using namespace std;
const int maxn=200005;
int n,m,k;
set<int> x[maxn];
set<int> y[maxn];
int main(){
	cin>>n>>m>>k;
	for(int i=1;i<=k;i++){
		int u,v;
		cin>>u>>v;
		x[u].insert(v);
		y[v].insert(u);
	}
	int Q;
	cin>>Q;
	while(Q--){
		int opt,p;
		cin>>opt>>p;
		if(opt==1){
			cout<<sz(x[p])<<endl;
			for(int i:x[p]){
				y[i].erase(p);
			}
			x[p].clear();
		}else{
			cout<<sz(y[p])<<endl;
			for(int i:y[p]){
				x[i].erase(p);
			}
			y[p].clear();
		}
	}
	return 0;
}

E. Popcount Sum 3

题意

多测，共 $T$ 组数据。

给你整数 $N$ 和 $K$ ，找到所有满足以下条件的数 $x$ 的和：

$\text{popcount}(x)=K$ 且 $ x\le N$，其中 $\text{popcount(x)}$ 表示 $x$ 在二进制形式中 $1$ 的数量

思路

提供一个不是 $\text{DP}$ 的做法。为了方便表示，设 $998244353 = M$ ；

对于 $N$ 二进制形式中每个是 $1$ 的位数 $d$ （从后往前数，最右边的是第 $1$ 位），把它之前（不包括自己）的所有是 $1$ 的位都默认设成 $1$ ，设之前有 $c n t$ 位是 $1$ （可以预处理 出前面是 $1$ 的数位表示的数的总和，即 $sum=\sum_{i=[x]在二进制形式下的位数}^{d+1} 2^{d-1}$ ），在之后（不包括自己）的共 $d - 1$ 位中任意放入 $K - c n t$ 个 $1$ ，很明显，方案数为 $\text{C}_{d-1}^{K-cnt}$ 。我们把这些方案全部列出来：

下图展示了 $k - c n t = 3, d - 1 = 5$ 的情况 ：

它是 $\text{C}_{d-1}^{K-cnt}$ 个长度为 $K - c n t$ 的序列，每个数在 $\sim d-1$ 之间，可以证明， $\sim d-1$ 之间的每一个数（每一位）出现的次数都一样，即每个数字（每一位）出现了 $t=\displaystyle \frac{\text{C}_{d-1}^{K-cnt} \times (K-cnt)}{d-1}$ 次：
第 $i$ 位表示的数为 $2^i$ ，则本次答案增加 $\Delta=\sum_{i=1}^{d-2}2^i\times t+\text{C}_{d-1}^{K-cnt}\times sum =\displaystyle \frac{\text{C}_{d-1}^{K-cnt} \times (K-cnt)}{d-1}\times (2^{d-1}-1)+\text{C}_{d-1}^{K-cnt}\times sum$ 。

如果 $c n t = K$ 就把答案加上 $s u m$ 并跳出循环。

注意点：

因为先取模再除法可能会出错，所以我们把除法换成乘法逆元，即 $\frac{1}{d-1} \bmod M=(d-1)^{M-2}\bmod M$ （可以用费马小定理证明）。
同理， $\text{C}_n^k$ 也需要预处理，预处理出 $fact_i=i! \bmod M$ ， $rev_i=\frac{1}{i!}\bmod M$ ， $\text{C}_n^k=fact_n\times rev_k\times rev_{n-k}$ 。
记得开 $\text{long long}$ ，计算幂要用快速幂，不然超时。

C++ 代码

#include<bits/stdc++.h>
#define int long long
#define mpr make_pair
#define pb push_back
#define sz(v) (int)v.size()
using namespace std;
const int MOD=998244353;
int n,k;
int fact[64],rev[64];
int pw(int a,int b){
	int res=1;
	while(b){
		if(b&1) res=res*a%MOD;
		a=a*a%MOD;
		b>>=1;
	}
	return res;
}
int C(int n,int k){
	if(n<k) return 0;
	return fact[n]*rev[k]%MOD*rev[n-k]%MOD;
}
void solve(){
	cin>>n>>k;
	vector<bool> v;
	while(n!=0){
		v.pb(n&1);
		n>>=1;
	}
	reverse(v.begin(),v.end());
	int ans=0,cnt=0,sum=0;
	for(int i=0;i<sz(v);i++){
		if(v[i]==0) continue;
		int dig=sz(v)-i;
		ans=(ans+(pw(2,dig-1)-1)*(C(dig-1,k-cnt)*(k-cnt)%MOD*pw(dig-1,MOD-2)%MOD)%MOD)%MOD;
		ans=(ans+C(dig-1,k-cnt)*sum)%MOD;
		cnt++;
		sum+=pw(2,dig-1);
		if(cnt==k){
			ans=(ans+sum)%MOD;
			break;
		}
	}
	cout<<ans<<endl;
}
signed main(){
	fact[1]=fact[0]=rev[0]=1;
	for(int i=2;i<=61;i++){
		fact[i]=fact[i-1]*i%MOD;
	}
	rev[61]=pw(fact[61],MOD-2);
	for(int i=60;i>=1;i--){
		rev[i]=rev[i+1]*(i+1)%MOD;
	}
	int T;
	cin>>T;
	while(T--){
		solve();
	}
	return 0;
}