2018.11.17 codechef PRIMEDST(点分治+fft)
传送门
fftfftfft一眼题(其实最先想到的是nttnttntt,wawawa了几次之后发现模数不够大果断弃疗写fftfftfft)
我们点分治统计答案的个数。
考虑现在已经统计出了到当前点的所有距离如何更新答案。
显然如果两个距离能够凑出一个质数的话就对答案有1的贡献。
所以相当于计算出每一个距离的个数。
然后可以推一个式子:
tota=∑i=0maxdiscnti∗cnta−itot_a=\sum_{i=0}^{maxdis}cnt_i*cnt_{a-i}tota=∑i=0maxdiscnti∗cnta−i
这不就是距离数组跟自己卷积吗?
果断fftfftfft优化。
然而fftfftfft的写法也wawawa了。
找dzyodzyodzyo拍了半天没错?
然后全局longlonglong longlonglong了一波终于过了233.
代码:
#include<bits/stdc++.h>
#define int long long
using namespace std;
inline int read(){
int ans=0;
char ch=getchar();
while(!isdigit(ch))ch=getchar();
while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
return ans;
}
typedef long long ll;
const double pi=acos(-1.0);
const int N=2e5+5;
int n,pri[N],tot=0,msiz[N],siz[N],vis[N],rt,all,tmp[N],lim=1,tim=0,pos[N],maxn;
bool isp[N];
vector<int>e[N>>2];
ll ans=0;
inline void init(){
isp[1]=1;
for(register int i=2;i<=n;++i){
if(!isp[i])pri[++tot]=i;
for(register int j=1;j<=tot&&i*pri[j]<=n;++j){
isp[i*pri[j]]=1;
if(i%pri[j]==0)break;
}
}
}
struct Complex{
double x,y;
friend inline Complex operator+(const Complex&a,const Complex&b){return (Complex){a.x+b.x,a.y+b.y};}
friend inline Complex operator-(const Complex&a,const Complex&b){return (Complex){a.x-b.x,a.y-b.y};}
friend inline Complex operator*(const Complex&a,const Complex&b){return (Complex){a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};}
friend inline Complex operator/(const Complex&a,const double&b){return (Complex){a.x/b,a.y/b};}
}cnt[N];
inline void fft(Complex a[],int type){
for(register int i=0;i<lim;++i)if(i<pos[i])swap(a[i],a[pos[i]]);
for(register int mid=1;mid<lim;mid<<=1){
Complex wn=(Complex){cos(pi/mid),sin(type*pi/mid)};
for(register int j=0,len=mid<<1;j<lim;j+=len){
Complex w=(Complex){1,0};
for(register int k=0;k<mid;++k,w=w*wn){
Complex a0=a[j+k],a1=a[j+k+mid]*w;
a[j+k]=a0+a1,a[j+k+mid]=a0-a1;
}
}
}
if(type==-1)for(register int i=0;i<lim;++i)a[i]=a[i]/(double)lim;
}
void getroot(int p,int fa){
msiz[p]=siz[p]=1;
for(register int i=0;i<e[p].size();++i){
int v=e[p][i];
if(v==fa||vis[v])continue;
getroot(v,p),siz[p]+=siz[v],msiz[p]=max(msiz[p],siz[v]);
}
msiz[p]=max(msiz[p],all-siz[p]);
if(msiz[p]<msiz[rt])rt=p;
}
void getdis(int p,int fa,int delt){
cnt[delt].x+=1;
maxn=max(maxn,delt);
for(register int i=0;i<e[p].size();++i){
int v=e[p][i];
if(vis[v]||v==fa)continue;
getdis(v,p,delt+1);
}
}
inline void calc(int p,int delt,int type){
maxn=0,getdis(p,0,delt),lim=1,tim=0;
while(lim<=maxn*2)++tim,lim<<=1;
for(register int i=0;i<lim;++i)pos[i]=(pos[i>>1]>>1)|((i&1)<<(tim-1));
int sum=(int)-cnt[1].x;
fft(cnt,1);
for(register int i=0;i<lim;++i)cnt[i]=cnt[i]*cnt[i];
fft(cnt,-1);
for(register int i=1;i<=tot;++i){
if(pri[i]>lim)break;
sum+=(int)(cnt[pri[i]].x+0.5);
}
ans+=(ll)sum/2*type;
for(register int i=0;i<lim;++i)cnt[i].x=cnt[i].y=0;
}
void solve(int p){
calc(p,0,1),vis[p]=1;
for(register int i=0;i<e[p].size();++i){
int v=e[p][i];
if(vis[v])continue;
all=siz[v],rt=0,calc(v,1,-1),getroot(v,0),solve(rt);
}
}
signed main(){
freopen("lx.in","r",stdin);
n=read(),init();
for(register int i=1,u,v;i<n;++i)u=read(),v=read(),e[u].push_back(v),e[v].push_back(u);
all=msiz[rt=0]=n,getroot(1,0),solve(rt);
printf("%.8lf",(double)ans*2.0/(double)((double)n*(double)(n-1)));
return 0;
}
2018.11.17 codechef PRIMEDST(点分治+fft)的更多相关文章
- International Programming Retreat Day(2018.11.17)
时间:2018.11.17地点:北京国华投资大厦
- 2018.11.17 bzoj4259: 残缺的字符串(fft)
传送门 fftfftfft套路题. 我们把aaa ~ zzz映射成111 ~ 262626,然后把∗*∗映射成000. 考虑对于两个长度都为nnn的字符串A,BA,BA,B. 我们定义一个差异函数di ...
- IOS面试题2018/11/17
1.设计模式是什么?你知道哪些设计模式? 设计模式是一种编码经验,就是一种成熟的逻辑去处理某一种类型的事情. 1.MVC模式:model view controller,把模型,视图,控制器 层进行解 ...
- 2018.11.17 hdu5829Rikka with Subset(ntt)
传送门 nttnttntt基础题. 考虑计算每一个数在排名为kkk时被统计了多少次来更新答案. 这样的话,设anskans_kansk表示所有数的值乘上排名为kkk的子集数的总和. 则ansk=∑i ...
- Django学习笔记-2018.11.17
URL配置: 项目下的urls.py配置的为URL总路径,在使用第二种方法在app下配置的urls.py是在总路径下的分路径 Templates: DTL初步使用 render()函数支持dict类型 ...
- 2018.11.17 Struts2框架入门
Struts2 框架学习 一.struts2是什么? (1)概念 (2)struts2使用优势 自动封装参数 参数校验 结果的处理(转发|重定向) 国际化 显示等待页面 表单的防止重复提交 (3)st ...
- 2017 3 11 分治FFT
考试一道题的递推式为$$f[i]=\sum_{j=1}^{i} j^k \times (i-1)! \times \frac{f[i-j]}{(i-j)!}$$这显然是一个卷积的形式,但$f$需要由自 ...
- hdu 5730 Shell Necklace [分治fft | 多项式求逆]
hdu 5730 Shell Necklace 题意:求递推式\(f_n = \sum_{i=1}^n a_i f_{n-i}\),模313 多么优秀的模板题 可以用分治fft,也可以多项式求逆 分治 ...
- 洛谷 P4721 【模板】分治 FFT 解题报告
P4721 [模板]分治 FFT 题目背景 也可用多项式求逆解决. 题目描述 给定长度为 \(n−1\) 的数组 \(g[1],g[2],\dots,g[n-1]\),求 \(f[0],f[1],\d ...
随机推荐
- Android 基础 (四大组件,五大存储,六大布局)
Android四大组件: 参考:https://blog.csdn.net/shenggaofei/article/details/52450668 Android四大组件分别为activity.se ...
- 【资料整理】ADO.NET
ADO.NET: 1.SYSTEM.DATA 命名空间下类的集合的统称,用于操作 连接数据库的 它提供了统一的编程接口,可以操作不同的数据库 数据库实例:默认实例(MSSQLSEVER)和命名实例(M ...
- 微信小程序——过滤器的模拟
>> 编写wxs文件——filter.wxs //1. 价格格式化 function getPriceFormat(value) { return parseFloat(isNaN(val ...
- 快速将磁盘的MBR分区方式改成GPT分区方式
1.按Shift + F10打开命令提示符. 2.diskpart 3.list disk(列出所有磁盘) 4.select disk 0(选择磁盘) 5.clean(格式化所选的磁盘) 7.conv ...
- composer 安装新包失败的原因之一
各种方法都尝试过了,然而最大的可能就是不能访问国外的资源! 1.使用vpnFQ下载 2.修改一下composer的配置,命令如下: composer config -g repo.packagist ...
- 如何查看Firefox中保存的登录密码
问:以前使用Firefox浏览器登录一个论坛,并且临时申请了一个账号,在使用Firefox登录时选择让它记住密码了,后来,我忘记了那个论坛的密码,但是可以使用Firefox直接登录.现在能不能查看密码 ...
- mysql 主从数据不一致 Slave_SQL_Running: No 解决方法
在slave服务器上通过如下命令 mysql> show slave status\G; 显示如下情况: Slave_IO_Running: Yes Slave_SQL_Running: No ...
- 在pom.xml中引入jar包坐标的依赖范围
A依赖B,需要在A的pom.xml文件中添加B的坐标,添加坐标时需要指定依赖范围,依赖范围包括: compile:编译范围,指A在编译时依赖B,此范围为默认依赖范围.编译范围的依赖会用在编译.测试.运 ...
- Winrar发现损坏的压缩文件头
解决方法: 点击"解压到"-->保留损坏文件
- UML 图C#
继承关系(类1继承类2) 代码: class Class1:Class2 { } class Class2 { } 实现(实现接口) 代码: interface interface1 { void s ...