首页 > 其他分享 >[题解]AT_abc215_g [ABC215G] Colorful Candies 2

[题解]AT_abc215_g [ABC215G] Colorful Candies 2

时间:2024-06-22 10:56:05浏览次数:27  
标签:ABC215G vis int 题解 复杂度 ++ Candies Theta binom

思路

定义 \(vis_i\) 表示数 \(i\) 在序列中出现的次数。如果我们选出 \(k\) 个数,答案就是(其中 \(m\) 表示 \(\max(c_i)\)):

\[ \sum_{i = 1}^m\frac{\binom{n}{x} - \binom{n - vis_i}{k}}{\binom{n}{x}} \]

显然,我们只枚举序列中存在的元素,时间复杂度 \(\Theta(n^2)\),过不了,考虑优化。

不难发现,对于答案的贡献与其权值无关,之和出现的次数有关。那么,对于所有满足 \(i \neq j \wedge vis_i = vis_j\) 的元素,对于答案的贡献都是一样的。因此将其看作一种元素考虑。

答案就转变为了(\(p\) 为压缩后的序列大小,\(a\) 为压缩后的序列):

\[ \sum_{i = 1}^p(cnt_i \times \frac{\binom{n}{k} - \binom{n - vis_{a_i}}{k}}{\binom{n}{k}}) \]

时间复杂度为 \(\Theta(np)\),因为在最坏情况下,出现次数分别是:\(1,2,3,\dots\)。所以 \(p\) 是 \(\Theta(\sqrt n)\) 级别的。

因此,时间复杂度为 \(\Theta(n \sqrt n)\)。

Code

#include <bits/stdc++.h>  
#define int long long  
#define re register  
  
using namespace std;  
  
const int N = 5e4 + 10,mod = 998244353;  
int n,m;  
int arr[N],brr[N],mul[N],inv[N];  
map<int,int> vis,mp;  
  
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 int exgcd(int a,int b,int &x,int &y){  
    if (!b){  
        x = 1;  
        y = 0;  
        return a;  
    }  
    int d = exgcd(b,a % b,y,x);  
    y = y - a / b * x;  
    return d;  
}  
  
inline void init(){  
    mul[0] = 1;  
    for (re int i = 1;i <= n;i++) mul[i] = mul[i - 1] * i % mod;  
    for (re int i = 0;i <= n;i++){  
        int a = mul[i],p = mod,x,y;  
        exgcd(a,p,x,y);  
        inv[i] = (x % mod + mod) % mod;  
    }  
}  
  
inline int C(int n,int m){  
    if (n < m) return 0;  
    return mul[n] * inv[n - m] % mod * inv[m] % mod;  
}  
  
signed main(){  
    n = read();  
    init();  
    for (re int i = 1;i <= n;i++){  
        int x;  
        x = read();  
        vis[x]++;  
    }  
    for (auto it = vis.begin();it != vis.end();it++) mp[it -> second]++;  
    for (auto it = mp.begin();it != mp.end();it++){  
        m++;  
        arr[m] = (it -> first);  
        brr[m] = (it -> second);  
    }  
    for (re int i = 1;i <= n;i++){  
        int ans = 0;  
        for (re int j = 1;j <= m;j++) ans = (ans + ((C(n,i) - C(n - arr[j],i)) % mod + mod) % mod * brr[j] % mod) % mod;  
        int a = C(n,i),p = mod,x,y;  
        exgcd(a,p,x,y);  
        int iv = (x % mod + mod) % mod;  
        printf("%lld\n",ans * iv % mod);  
    }  
    return 0;  
}  

标签:ABC215G,vis,int,题解,复杂度,++,Candies,Theta,binom
From: https://www.cnblogs.com/WaterSun/p/18261946

相关文章

  • [题解]AT_abc195_d [ABC195D] Shipping Center
    思路一个简单的贪心,对于每一次操作,我们假设我们能用盒子的大小的数组处理成\(a\)。那么,我们可以对\(a\)进行从小到大排序。然后,对于我们所有的箱子,我们可以以\(w\)为关键字,从小到大排序。接着,我们可以进行暴力枚举,对于\(a_i\),我们要取的必定为\(\max_{w_j\leqa_i}(v_j......
  • [题解]AT_abc158_e [ABC158E] Divisible Substring
    思路首先发现一个事情,任意一个子串都可以由\(s\)的某一个后缀的后面删除一些字符得到。因此假如\(s\)的某一个后缀的值为\(x\),那么我们可以减去后面的我们不用的数字\(a\),然后除以\(10\)的若干次幂得到,即\(\frac{x-a}{10^n}\)。于是得到:\[\frac{x-a}{10^n}\equi......
  • [题解]AT_abc153_f [ABC153F] Silver Fox vs Monster
    模拟赛最后\(15\)分钟想到的做法。思路首先有一个显然的贪心策略:我们放炸弹的地方要尽可能的使这个炸弹能影响到更多的怪上。那么我们可以将对于一个怪\(i\)能够影响到它的区间表示出来\([\max(1,l_i-d),a_i+r]\)。然后将这些区间排个序,可以粗略画出这样的图:根据上......
  • [题解]AT_abc151_e [ABC151E] Max-Min Sums
    思路考虑将\(\max\)和\(\min\)的贡献分开计算。显然我们对这个序列进行一次排序不会影响最终的答案,因此我们可以先排序一下。然后有一个很经典的trick,就是你枚举每一个数\(x\),将\(x\)令为最大值(最小值)。因为我们先前排序过一次,因此我们可以轻易的计算出比\(x\)小(大)的......
  • [题解]AT_abc236_e [ABC236E] Average and Median
    思路直接将输出的答案分为两个分考虑。(1)考虑二分+DP。设当前二分出的平均数为\(x\),如果合法,那么有(其中\(p\)为选出数下标的集合):\[\frac{a_{p_1}+a_{p_2}+\dots+a_{p_k}}{k}\geqx\]即:\[\frac{(a_{p_1}-x)+(a_{p_2}-x)+\dots+(a_{p_......
  • 2020C++等级考试二级真题题解
     202012数组指定部分逆序重放c++ #include<iostream>usingnamespacestd;intmain(){  inta[110];  intn,k;  cin>>n>>k;  for(inti=0;i<n;i++){    cin>>a[i];  }  for(inti=0;i<k/2;i++){......
  • 题解:P10641 BZOJ3252 攻略
    我让cz搬这道题,cz给搬了,于是来写个题解(考虑一个朴素的贪心:每次选择一个到根路径价值和最大的叶子,将价值和累加进答案,并把这条链价值清零。这个贪心的正确性显然(可以交换法证明),很容易用数据结构维护做到\(O(n\logn)\)。但是这样太不优美了,而且数据结构比较难写,于是考虑一个......
  • P8500 [NOI2022] 冒泡排序 题解
    考虑特殊性质B。限制相当于钦定一些位置的值,其他位置无限制。可以发现性质:无限制的位置上填的值是单调不减的。证明:设得到的最优序列为\(A\),对于无限制的位置\(i,j\),若\(A_i>A_j\),交换\(i,j\)后逆序对个数必然减小。根据改性质,只需考虑每个位置对已经确定位置的位置的贡......
  • P4317 花神的数论题 题解
    头话说好久没写题解了P4317花神的数论题题链题意:给你一个不超过\(10^{15}\)的数\(n\),求\(\prod_{i=1}^nsum_i\),其中\(sum_i\)表示\(i\)在二进制表示下\(1\)的个数。学了几道题后,本能的设出了\(f_{i,j}\)表示\(i\)位数中含\(j\)个\(1\)的数的个数,转移......
  • 【题解】P6323 | 容斥 分拆数
    本题存在低于\(O(nc)\)的做法。逆序对是大小关系,我们在小的那个数处统计每对逆序对,考虑从大到小插入每一个数,这样所有数都比他大,这样它插入在第\(i\)个就会产生\(i\)个逆序对,假设现在有\(x\)个数则它可以产生\([0,x]\)中个逆序对,且每种都恰好有一种插法。那么我们现在......