D

小红的整数转换

https://ac.nowcoder.com/acm/contest/71593/A

代码写的比较丑,这道题的思路就是先用不管取不取红蓝硬币,先算出混取凑出面值为p的方案数,然后算出只取红硬币凑出的面值方案数,最后算出只取蓝硬币凑出的方案数,最后就是把混取面值为p的方案数减去只取红或蓝得到面值为p的方案数,这样得到的一定就是有红有蓝的方案了,最后取模就好了

int fa[MAXN],fb[MAXN],fc[MAXN];
void solve() {
	int n,p;cin>>n>>p;
    string str;cin>>str;str=' '+str;
    for(int i=1;i<=n;i++) cin>>c[i];
    vector<int>va,vb;
    for(int i=1;i<=n;i++){
        if(str[i]=='B')  vb.push_back(i);
        else va.push_back(i);
    }
    int ida=1;
    for(auto it:va) a[ida]=c[it],ida++; ida--;
    int idb=1;
    for(auto it:vb) b[idb]=c[it],idb++; idb--;
    fa[0]=1;
    for(int i=1;i<=ida;i++){
        for(int j=p;j>=a[i];j--){
        	fa[j]=(fa[j]+fa[j-a[i]])%mod;
        }
    }
    fb[0]=1;
    for(int i=1;i<=idb;i++){
        for(int j=p;j>=b[i];j--){
        	fb[j]=(fb[j]+fb[j-b[i]])%mod;
        }
    }
    fc[0]=1;
    for(int i=1;i<=n;i++){
        for(int j=p;j>=c[i];j--){
        	fc[j]=(fc[j]+fc[j-c[i]])%mod;
        }
    }
    cout<< (((fc[p]-fb[p]-fa[p])%mod)+mod)%mod <<endl;
}
全部评论
不知道如果n种颜色这题应该怎么思考呢
点赞 回复 分享
发布于 2023-12-10 20:57 湖北

相关推荐

09-01 11:31
门头沟学院 Java
buul:七牛云的吧,感觉想法是好的,但是大家没那么多时间弄他这个啊。。。不知道的还以为他是顶尖大厂呢还搞比赛抢hc,只能说应试者的痛苦考察方是无法理解的,他们只会想一出是一出
点赞 评论 收藏
分享
09-01 09:00
已编辑
四川旅游学院 运营
牛客55195891...:主要是专业不好,别的没毛病
牛客解忧铺
点赞 评论 收藏
分享
评论
3
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务