牛客小白月赛43-E

2022-01-21 14:31:06 阅读：175 来源： 互联网

满意的集合

https://ac.nowcoder.com/acm/contest/11220/E

题意：

有数字 1∼∼9，每个数字的个数分别为 cnt1,cnt2,cnt3,...,cnt9。计算出“满意的集合“的个数。

"满意的集合" 定义：选出的数存在一种排列方式，其拼接起来后表示的十进制整数，能被 3整除，例如集合 {3,3,6} (包含了 2 个数字 3,1 个数字 6 )，可以有排列 {6,3,3} 代表十进制下的整数 633，能被 3 整除。链

两个集合相同，当且仅当集合元素个数相同，且排序后对应数字相同，例如 {3,3,6}和 {3,6,3}是同样的集合。空集合看作 0 ，是合法的，答案对 1e9+7取模。

思路：

一个十进制数每一位累加是 3 的倍数，那么这个数必然会被 3整除

暴力做法
for (int i=1; i<=9; i++) {
    for (int j=0; j<=cnt[i]; j++) {
        for (int MOD=0; MOD<3; MOD++) {
            dp[i][(j*i+MOD)%3]+=dp[i-1][MOD];
            dp[i][(j*i+MOD)%3]%=mod;
        }
    }
}
由于数据高达 1e9不可行

仔细观察发现，对于一个数 i ，选了 1,4,7,..,3∗x+1个的时候，得到的关于 3 的模数都是相等的，

即1∗i % 3=4∗i % 3=7∗i % 3=...=(3∗x+1)∗i % 3

同理

2∗i % 3=5∗i % 3=8∗i % 3=...=(3∗x+2)∗i % 3

0∗i % 3=3∗i % 3=6∗i % 3=...=(3∗x)∗i % 3

因此只有 3 个可能的模数，复杂度可以优化成 O (9∗3∗3)。

代码：

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn=5e5+10;
const int mod=1e9+7;
int a[20];
int dp[10][5];//表示只选择数字 1-i ，累加和 % 3 =j 的方案有多少
/*
for (int i=1; i<=9; i++) {
    for (int j=0; j<=cnt[i]; j++) {
        for (int MOD=0; MOD<3; MOD++) {
            dp[i][(j*i+MOD)%3]+=dp[i-1][MOD];
            dp[i][(j*i+MOD)%3]%=mod;
        }
    }
}
暴力做法


*/
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    for(int i=1;i<=9;i++){
        cin>>a[i];
    }
    dp[0][0]=1;
    for(int i=1;i<=9;i++){
        int a1=i%3,a2=i*2%3,a3=i*3%3;
        int b1=(a[i]+2)/3,b2=(a[i]+1)/3,b3=a[i]/3+1;
        for(int k=0;k<3;k++){
            dp[i][(a1+k)%3]+=b1*dp[i-1][k];
            dp[i][(a1+k)%3]%=mod;
            dp[i][(a2+k)%3]+=b2*dp[i-1][k];
            dp[i][(a2+k)%3]%=mod;
            dp[i][(a3+k)%3]+=b3*dp[i-1][k];
            dp[i][(a3+k)%3]%=mod;
        }
    }
    cout<<dp[9][0]<<endl;
	system("pause");

}
/*
一个十进制数每一位累加是 3 的倍数，那么这个数必然会被 3整除。

*/

标签：...,数字,int,43,牛客,小白月赛,集合,整除,十进制
来源： https://www.cnblogs.com/CurryBP/p/15830106.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

牛客小白月赛43-E

满意的集合

题意：

思路：

代码：