ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

【狄利克雷前缀和 / 后缀和】算法学习

2022-09-11 23:00:28  阅读:206  来源: 互联网

标签:ch 前缀 狄利克 后缀 int seed alpha include define


1. 狄利克雷前缀和

问题描述

有数列 \(\{a\}\),求数列 \(\{b\}\) 满足

\[b_k = \sum_{i|k} a_i \]

数列长度 \(n \le 2 \times 10 ^ 7\)。

分析

考虑质因数分解,某个数 \(x = \prod\limits p_{i} ^ {\alpha_i}\), 将其写成行向量 \((\alpha_1,\alpha_2, \dots,\alpha_k)\)。 那么每次乘一个素数 \(p_i\),使得 \(\alpha_i + 1\)。 于是可以用这个向量描述贡献 \(x \to x\cdot p_i\) 。这本质上是个 \(k\) 维前缀和的过程。

可是素数是无限多的,我们考虑状态压缩,发现只需要用原来的数表示这个向量即可。状态数 \(O(n)\)。

P5495
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cmath>
#include <queue>
typedef long long ll;
typedef double db;
using namespace std;
#define PII pair<int,int>
#define mp make_pair
#define fir first
#define sec second
#define fin(haha) freopen(#haha".in","r",stdin);
inline int read(){
	register int x=0,f=0,ch=getchar();
	while('0'>ch||ch>'9')f^=ch=='-',ch=getchar();
	while('0'<=ch&&ch<='9')x=x*10+(ch^'0'),ch=getchar();
	return f?-x:x;
}

const int MAX = 2e7 + 5;
#define uint unsigned int
uint seed;
inline uint getnext(){
	seed^=seed<<13;
	seed^=seed>>17;
	seed^=seed<<5;
	return seed;
}
uint a[MAX], b[MAX];
int p[MAX], f[MAX];
void init(int N) {
	for(int i=2;i<=N;++i) {
		if(!f[i]) f[i] = 1,p[++p[0]] = i;
		for(int j=1;j<=p[0] && i*p[j]<=N;++j) {
			f[i*p[j]] = 1;
			if(i % p[j] == 0)break;
		}
	}
}
signed main(){
	int n = read(); cin>>seed;
	for(int i=1;i<=n;++i) a[i] = getnext();
	init(n);
	for(int i=1;i<=p[0];++i) {
		for(int j=1; j*p[i] <= n;++j) {
			a[j * p[i]] += a[j];
		}
	}
	uint ans = 0;
	for(int i=1;i<=n;++i) ans ^= a[i];
	cout << ans << endl;
	return 0;
}

标签:ch,前缀,狄利克,后缀,int,seed,alpha,include,define
来源: https://www.cnblogs.com/Lates/p/16685088.html

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

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有