求相同特征量题型

向夕阳Salute

已于 2024-03-01 16:28:58 修改

阅读量71

点赞数 1

分类专栏：算法文章标签： c++ 算法图论

于 2023-04-12 09:15:57 首次发布

本文链接：https://blog.csdn.net/qq_63092029/article/details/130098699

版权

算法专栏收录该内容

95 篇文章 0 订阅

订阅专栏

文章介绍了如何通过设置状态量来计算字符串中特定字符（如0和1）重复出现的子串数量。提供了两种情况的代码示例，一种是处理两个字符相同的情况，另一种是处理三个字符相同的情况，利用动态规划方法更新状态并计算结果。

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

找包含特征量相同的字串数量

$技巧$
$可以设置一个状态量，表示当前每个特征字符的数量$
$相当于一个 i 一直向后推进，然后一直加符合题意的状态量$

两个字符相同的例子

//找字串中'0','1'字符相同的字串数量
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int n;
string s;
int cnt[N];
int main()
{
    cin >> n;
    cin >> s;
    int res = 0;
    int cnt_1 = 0;
    cnt[0] = 1;
    for(auto t : s)
    {
        if(t == '1')
        {
            cnt_1 ++;
            res += cnt[cnt_1];
            cnt[cnt_1] ++;
        }
        else
        {
            cnt_1 --;
            res += cnt[cnt_1];
            cnt[cnt_1] ++;
        }
    }
    cout << res << endl;

    return 0;
}

三个字符相同的例子

链接牛客例题

#include<bits/stdc++.h>
using namespace std;
const int N = 3e5 + 10;
int t, n;
map<pair<int, int>, int> cnt;
int main()
{
    cin >> t;
    while(t --)
    {
        long long res = 0;
        cnt.clear();
        cnt[{0,0}] = 1;
        int cnt_0 = 0;
        int cnt_1 = 0;
        cin >> n;
        string s;
        cin >> s;
        for(auto t: s)
        {
            if(t == '1')
            {
                cnt_1 ++;
                res += cnt[{cnt_0, cnt_1}];
                cnt[{cnt_0, cnt_1}] ++;
            }
            else if(t == '0')
            {
                cnt_0 ++;
                res += cnt[{cnt_0, cnt_1}];
                cnt[{cnt_0, cnt_1}] ++;
            }
            else{
                cnt_0 --;
                cnt_1 --;
                res += cnt[{cnt_0, cnt_1}];
                cnt[{cnt_0, cnt_1}] ++;
            }
        }
        cout << res << endl;
    }
    return 0;
}