AtCoder Beginner Contest 169(题解)

E - Count Median

结论题

给定 n n n x i ∈ [ a i , b i ] x_i\in[a_i,b_i] xi[ai,bi],求中位数的个数。

定义: k = ⌊ n 2 ⌋ k=\lfloor\dfrac{n}{2}\rfloor k=2n,对 a , b a,b a,b进行排序后,为 a k + 1 a_{k+1} ak+1 a i a_i ai的中位数, b k + 1 b_{k+1} bk+1 b i b_i bi的中位数。

1. n n n为奇数,则范围是 [ a k + 1 , b k + 1 ] [a_{k+1},b_{k+1}] [ak+1,bk+1]

2. n n n为偶数,定义则范围是 [ a k + a k + 1 , b k + b b + k + 1 ] [a_{k}+a_{k+1},b_k+b_{b+k+1}] [ak+ak+1,bk+bb+k+1]


F - Knapsack for All Subsets

转移与物品个数有关的背包问题

考虑令 d p [ i ] [ j ] dp[i][j] dp[i][j] i i i个物品答案为 j j j的方案数。目标是 d p [ n ] [ s ] dp[n][s] dp[n][s]

初始 d p [ 0 ] [ 0 ] = 1 × 2 n − 0 = 2 n dp[0][0]=1\times 2^{n-0}=2^n dp[0][0]=1×2n0=2n

有转移方程: d p [ i ] [ j ] = ( d p [ i ] [ j ] + d p [ i − 1 ] [ j − a [ i ] ] 2 ) dp[i][j]=(dp[i][j]+\dfrac{dp[i-1][j-a[i]]}{2}) dp[i][j]=(dp[i][j]+2dp[i1][ja[i]])

因为每多选一个物品,剩余未选的物品数就少一,而方案数与 2 n − k 2^{n-k} 2nk有关, k k k为子集大小。

可以利用滚动数组倒序,优化到一维。

c o d e code code

// Problem: F - Knapsack for All Subsets
// Contest: AtCoder - AtCoder Beginner Contest 169
// URL: https://atcoder.jp/contests/abc169/tasks/abc169_f
// Memory Limit: 1024 MB
// Time Limit: 2000 ms
// Date: 2021-03-12 20:17:55
// --------by Herio--------

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 
const int N=3e3+5,M=2e4+5,inf=0x3f3f3f3f,mod=998244353;
#define mst(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define fi first
#define se second
#define pb emplace_back
#define SZ(a) (int)a.size()
#define IOS ios::sync_with_stdio(false),cin.tie(0) 
void Print(int *a,int n){
	for(int i=1;i<n;i++)
		printf("%d ",a[i]);
	printf("%d\n",a[n]); 
}
int n,s,a[N];
ll ksm(ll a,ll n,int m=mod){
	ll ans=1;
	while(n){
		if(n&1) ans=ans*a%m;
		a=a*a%m;
		n>>=1;
	}
	return ans;
}
ll dp[N];
int main(){
	scanf("%d%d",&n,&s);
	ll inv2=ksm(2,mod-2);
	dp[0]=ksm(2,n);
	//printf("--%lld %lld\n",dp[0],inv2);
	for(int i=1;i<=n;i++){
		int x;scanf("%d",&x);
		for(int j=s;j>=x;j--)
			dp[j]=(dp[j]+dp[j-x]*inv2%mod)%mod;
	}printf("%lld\n",dp[s]);
	return 0;
}