Codeforces1536 C. Diluc and Kaeya（思维）

最新推荐文章于 2022-07-07 16:21:59 发布

live4m

最新推荐文章于 2022-07-07 16:21:59 发布

阅读量298

点赞数 1

本文链接：https://blog.csdn.net/weixin_44178736/article/details/117870580

版权

本文介绍了一种解决字符串中'D'和'K'字符分布问题的算法，通过计算它们的数量比例并应用递归，找到最优拆分使得子问题中'd/k'保持不变。核心代码展示了如何使用map记录不同比例的位置并返回最大出现次数作为答案。

摘要由CSDN通过智能技术生成

题意：

在这里插入图片描述

解法：

首先对于前缀[1,i],计算D的数量d和K的数量k,当前d和k的数量比为d/k.
由于每个部分d/k相同,那么你拆分出一段之后,剩下数量中d/k是不变的,那么就变为了一个子问题.

我们要令子问题递归的次数尽可能多,
用map标记d/k的位置有多少个,这里d/k要化为最简分数,d/k的最大长度(即出现次数)就是当前答案.

code：

#include<bits/stdc++.h>
#define int long long
#define PI pair<int,int>
using namespace std;
const int maxm=2e6+5;
char s[maxm];
int ans[maxm];
int n;
void solve(){
    cin>>n;
    cin>>(s+1);
    int c=0,cc=0;
    map<PI,int>mp;
    for(int i=1;i<=n;i++){
        if(s[i]=='D')c++;
        else if(s[i]=='K')cc++;
        int g=__gcd(c,cc);
        mp[{c/g,cc/g}]++;
        ans[i]=mp[{c/g,cc/g}];
    }
    for(int i=1;i<=n;i++){
        cout<<ans[i]<<' ';
    }
    cout<<endl;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    int T;cin>>T;while(T--)
    solve();
    return 0;
}