首页 > 其他分享 >[题解]AT_agc054_b [AGC054B] Greedy Division

[题解]AT_agc054_b [AGC054B] Greedy Division

时间:2024-06-23 13:12:14浏览次数:19  
标签:arr int 题解 sum AGC054B agc054 ans dp mod

思路

首先不难发现一个规律,当 \(sum\) 为奇数时不可能有解。

定义 \(dp_{i,j,k,0/1}\) 表示 A 在前 \(i\) 个数中选出和为 \(j\) 的 \(k\) 个数,且第 \(i\) 个 不选/选 的方案数。

那么,我们只需要对于第 \(i\) 个数的状态分类讨论就能得到状态转移方程:

  1. 不选 \(i\),\(dp_{i,j,k,0} = dp_{i - 1,j,k,0/1}\)。
  2. 选 \(i\),\(dp_{i,j,k,1} = dp_{i - 1,j - a_i,k - 1,0/1}\)。

综上,得到状态转移方程:

\[ \left\{\begin{matrix} dp_{i,j,k,0} = dp_{i - 1,j,k,0/1}\\ dp_{i,j,k,1} = dp_{i - 1,j - a_i,k - 1,0/1} \end{matrix}\right. \]

然后,我们就得到了序列 \(a\) 固定时的答案。那么,我们还要考虑一下 \(a\) 不固定的情况。

假设 A 最后取出了 \(x\) 个数,那么,我们就可以就可以对 A 和 B 两人选取数的序列进行全排列,即 \((x!)((n - i)!)\)。

现在还有一个问题,为什么不能直接对 \(a\) 进行全排列,也就是直接乘 \(n!\)。那是因为如果 A 和 B 两人选取的序列是固定的,由于题目中的条件当前权值小的人才能取数,所以说当两人选取数字的序列固定了,那么,拼起来的序列 \(a\) 也是固定的。

综上,答案为 \(\sum_{i = 1}^{n - 1}(dp_{n,\frac{sum}{2},i,0/1} \times (i!) \times ((n - i)!))\)。

这样就可以得到 MLE 1 个点 的代码。

然后,我们发现 \(dp_i\) 时的所有状态都是由 \(dp_{i - 1}\) 转移来的,所以说直接滚动一下即可。

Code

#include <bits/stdc++.h>  
#define int long long  
#define re register  
  
using namespace std;  
  
const int N = 110,M = 1e4 + 10,mod = 998244353;  
int n,r,sum,ans;  
int arr[N],mul[N];  
int dp[2][M][N][2];  
  
inline int read(){  
    int r = 0,w = 1;  
    char c = getchar();  
    while (c < '0' || c > '9'){  
        if (c == '-') w = -1;  
        c = getchar();  
    }  
    while (c >= '0' && c <= '9'){  
        r = (r << 3) + (r << 1) + (c ^ 48);  
        c = getchar();  
    }  
    return r * w;  
}  
  
inline void init(){  
    mul[0] = 1;  
    for (re int i = 1;i <= n;i++) mul[i] = mul[i - 1] * i % mod;  
}  
  
signed main(){  
    n = read();  
    init();  
    for (re int i = 1;i <= n;i++){  
        arr[i] = read();  
        sum += arr[i];  
    }  
    if (sum & 1){  
        puts("0");  
        return 0;  
    }  
    dp[0][0][0][0] = dp[0][arr[1]][1][1] = 1;  
    for (re int i = 2;i <= n;i++){  
        r ^= 1;  
        for (re int j = 0;j <= sum;j++){  
            for (re int k = 0;k <= i;k++) dp[r][j][k][0] = dp[r][j][k][1] = 0;  
        }  
        for (re int j = 0;j <= sum;j++){  
            for (re int k = 0;k <= i;k++){  
                dp[r][j][k][0] = (dp[r ^ 1][j][k][0] + dp[r ^ 1][j][k][1]) % mod;  
                if (j >= arr[i]) dp[r][j][k][1] = (dp[r ^ 1][j - arr[i]][k - 1][0] + dp[r ^ 1][j - arr[i]][k - 1][1]) % mod;  
            }  
        }  
    }  
    for (re int i = 1;i < n;i++) ans = (ans + (dp[r][sum / 2][i][0] + dp[r][sum / 2][i][1]) % mod * mul[i] % mod * mul[n - i] % mod) % mod;  
    printf("%lld",ans);  
    return 0;  
}  

标签:arr,int,题解,sum,AGC054B,agc054,ans,dp,mod
From: https://www.cnblogs.com/WaterSun/p/18263290

相关文章

  • [题解]AT_abc350_g [ABC350G] Mediator
    思路有加边操作,一眼LCT。问题在于处理询问操作。首先,判断联通。如果\(x,y\)不在同一个联通块内,则一定没有答案。其次,求出\(x,y\)之间节点的数量\(num\)(包括\(x,y\))。如果\(num=3\)说明\(x,y\)之间有一个共同的节点;如果\(num=2\)说明\(x,y\)直接连接;如果\(n......
  • [题解]AT_abc343_g [ABC343G] Compress Strings
    思路首先假设有两个串\(a,b\),如果\(b\)是\(a\)的子串,且\(a\neqb\)则不需要考虑\(b\);如果\(a=b\),则如需要保留一个\(a\)。做完上述操作后,显然最终的答案是由这些串按照一定顺序拼接起来,再删掉重叠部分。例如:abbcc与ccdde拼接为abbccccdde,发现cc是重复的,所以......
  • [题解]AT_abc342_f [ABC342F] Black Jack
    思路发现自己与庄家的操作是完全独立的,所以考虑分别计算它们。首先考虑自己的情况,定义\(dp_i\)表示掷出骰子的和为\(i\)获胜的概率,并记\(f(i)\)表示\(x=i\)时就不掷的获胜概率。对于每一步我们要么掷骰子(并且掷出的值等概率的在\(1\simD\)中),要么直接结束。两种情......
  • [题解]CF855E Salazar Slytherin's Locket
    思路毒瘤数位DP题。首先,你可以用一个vector储存每一个数字出现的次数,然后用map记忆化。然后可以得到如下TLE#8的代码。因为map自带一只\(\log\)所以,考虑将map优化掉。但是,现在每一种数字可能会出现很多次,所以要用vector维护出现次数,但这样必定需要用map一......
  • [题解]CF666B World Tour
    CSP-2022S2T1弱化版。思路首先因为边权均为\(1\),所以我们可以在\(\Theta(n^2)\)的复杂度用BFS求解出任意两点\(i,j\)的最短距离\(d_{i,j}\)(如果\(i\)不能到达\(j\),则令\(d_{i,j}=-1\))。有一个贪心的结论,就是使每一条\(A\toB,B\toC,C\toD\)的路径长度......
  • [题解]CF622F The Sum of the k-th Powers
    思路首先发现\(\sum_{i=1}^{n}i^k\)是一个\(k+1\)次多项式,那么我们需要求出\(k+2\)个点才能得到唯一的一个\(f(t)=\sum_{i=1}^{t}{i^k}\)。不难通过拉格朗日插值法,将\(x=1\sim(k+2)\)的情况一一带入:\[f(n)=\sum_{i=1}^{k+2}{((\sum_{j=1}^{i}......
  • [题解]CF622D Optimal Number Permutation
    思路首先考虑答案下界,因为\((n-i)\)和\(|d_i+i-n|\)均大于等于\(0\),所以它们相乘一定大于等于\(0\)。于是考虑能不能构造出结果为\(0\)。显然当\(i=n\)时,无论\(d_i\)的值是什么,式子的结果为\(0\)。因此只需要考虑\(i\in[1,n)\)的情况。因为要使结果为......
  • [题解]CF609F Frogs and mosquitoes
    思路发现\(x\)对题目的限制较大,因此考虑先将\(x\)排序并离散化后再来考虑。不难用线段树维护\(\max_{i=l}^{r}\{x_i+len_i\}\),这样我们就可以利用类似线段树上二分的技巧得出是哪一只青蛙吃掉的蚊子。但是有可能有一只蚊子无法吃掉,我们先把它丢进一个集合里面。每有......
  • [题解]CF988D Points and Powers of Two
    思路首先发现选出的数最多\(3\)个,考虑反证法。假设选出了四个数\(a,b,c,d\),并令:\[|a-b|=2^{x_1},|b-c|=2^{x_2},|c-d|=2^{x_3}\]又因为,\(|a-c|,|b-d|\)也都是\(2\)的次幂,那么有\(x_1=x_2=x_3\)。于是\(|a-d|=3\times2^{x_0}\neq2^k\)。在......
  • P9999 [Ynoi2000] tmostnrq 题解
    巨大难写题。就这样一个毒瘤的题,还有人把时空缩小成二分之一放在模拟赛,太好笑了。思路首先将询问离线。我们在\(l_i\)处加入这个点,在\(r_i\)处查询这个点在哪里。那么我们就需要有一个数据结构支持让所有树上的节点一起动。考虑所有点往\(x\)处动。那么对于在\(1\si......