B. Hamming Distance Sum（思维）

to cling

已于 2022-06-05 13:22:29 修改

阅读量189

点赞数

分类专栏：思维

于 2021-09-09 22:07:11 首次发布

本文链接：https://blog.csdn.net/QQ2530063577/article/details/120210711

版权

哈夫曼距离字符串子串前缀和优化算法 01串

关键词由CSDN通过智能技术生成

思维专栏收录该内容

22 篇文章 0 订阅

订阅专栏

传送门

题意：定义两个串的哈夫曼距离距离为
现在给定两个01串a，b（|a| <= |b|），求出a串与b的所有子串（连续的子串）的哈夫曼距离之和。
思路：设a和b的长度分别n和m。直接暴力的话是O（n²）算法，肯定不行。所以需要降低复杂度。可以分别考虑a串中每个字符对于答案的贡献度。提到这里顺便提一下是利用前缀和思想。若还想不明白，就再提示一下。设a中第i个字母的下标为i（从1开始），则第i个字符的贡献度为
a[i]分别与b中第i个字符到第n - m + i个字符的差求和。具体看代码。

#include<cstdio>
#include<queue>
#include<set>
#include<cstdlib>
#include<string.h>
#include<string>
#include<iostream>
#include<cmath>
#include<map>
#include<algorithm>
#define endl "\n"
#define IOS ios::sync_with_stdio(0), cin.tie(0),cout.tie(0)
#define ft first
#define sd second
#define pll pair<ll, ll>
#define pii pair<int, int>
#define ll long long int
#define mt(a,b) memset(a, b, sizeof a)
//#define int long long
const int inf = 0x3f3f3f3f;
const int INF = 0x7fffffff;
using namespace std;
const int N = 2e5 + 7, M = 1e6;
ll num[N];
int main()
{
	IOS;
	string a, b;
	cin >> a >> b;

	ll sum = 0;//注意，结果会爆int
	ll n = a.size(), m = b.size();
	a = " " + a; b = " " + b;//下标从1开始，便于做前缀和处理
	for (int i = 1; i <= m; i++)//b前缀和，也即1的个数
		if (b[i] == '1') num[i] = num[i - 1] + 1;
		else num[i] = num[i - 1];
	
	for (int i = 1; i <= n; i++)
	{
		//若为前者1则是后者中0的个数
		if (a[i] == '1') sum += m - n + 1 - (num[i + m - n] - num[i - 1]);
		else sum += num[i + m - n] - num[i - 1];
		//若为0，则为1的个数。
	}

	cout << sum << endl;

	return 0;
}