点分治详解:大佬的博客
P3806 【模板】点分治1
模板代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define inf 999999999
int n,m,len=0,Size;
struct node{int x,y,z,next;};
node e[20010];
int first[10010];
int root,ms,size[10010],mson[10010],sum[10000010];
bool v[10010];
void buildroad(int x,int y,int z)
{
len++;
e[len].x=x; e[len].y=y; e[len].z=z;
e[len].next=first[x];
first[x]=len;
}
//初始化以x为节点的子树
void getroot(int x,int fa)//fa表示x的父亲,防止往回搜
{
size[x]=1;mson[x]=0;
//size[i]记录以i为根的子树的大小,mson[i]表示i节点的最大子树的大小
for(int i=first[x];i;i=e[i].next)
{
int y=e[i].y;
if(v[y]||y==fa)continue;
//v[y]表示当前节点是否被分治过
getroot(y,x); //往下继续搜索
size[x]+=size[y]; //加上子树大小
if(size[y]>mson[x]) mson[x]=size[y]; //更新最大的子树
}
if(Size-size[x]>mson[x]) mson[x]=Size-size[x];
//Size表示当前这整棵树的大小,那么Size-size[x]就表示不在x的子树内的节点数量,下面详解
if(ms>mson[x]) ms=mson[x],root=x;
//ms表示树的重心的最大子树的大小(相当于mson[root]),这一步是用来更新树的重心的,root用来记录重心
}
int t; //临时变量,记录dis的编号
int dis[10010];
int ask[110],ans[110];
//初始化每个节点到根节点x的距离
void getdis(int x,int fa,int z)//fa表示x的父亲,z表示x到目标点的距离
{
dis[++t]=z;
//这里写dis[++t]=z而不是dis[x]=z是因为只需要求得匹配数,所以我们只需要知道每一个点到目标点的距离,而不需要知道那个点是谁,
for(int i=first[x];i;i=e[i].next)
{
int y=e[i].y;
if(y==fa||v[y])continue;
//这里加个v[y]是因为不能往被分治过的点走,分治过的点就没了
getdis(y,x,z+e[i].z);
}
}
struct asd{int x,y;}arr[10010];
int tt;
bool cmp(int x,int y){return x<y;}
//x表示getdis的起始点,y表示x到目标点的距离,id表示这一次统计出来的答案是合理的还是不合理的
void solve(int x,int y,int id)
{
t=0;
getdis(x,0,y);//算出树中的点到目标点的距离
tt=0;
sort(dis+1,dis+t+1,cmp);
dis[0]=-233;
//合并相同距离的点,并记录数目
for(int i=1;i<=t;i++)
if(dis[i]!=dis[i-1])arr[++tt].x=dis[i],arr[tt].y=1;
else arr[tt].y++;
for(int i=1;i<=m;i++)
{
//特判距离为k/2的点
if(ask[i]%2==0)for(int j=1;j<=tt;j++)
if(arr[j].x==ask[i]/2)ans[i]+=(arr[j].y-1)*arr[j].y/2*id;
//二分寻找其他满足距离和为k的点对,暴力求解复杂度为O(n^2),所以优化一下,枚举小于k/2的点,然后二分大于k/2的点,复杂度O(nlogn)
for(int j=1;j<=tt&&arr[j].x<ask[i]/2;j++)
{
int l=j+1,r=tt;
while(l<=r)
{
int mid=l+r>>1;
if(arr[j].x+arr[mid].x==ask[i])
{
ans[i]+=arr[j].y*arr[mid].y*id;
break;
}
if(arr[j].x+arr[mid].x>ask[i])r=mid-1;
else l=mid+1;
}
}
}
}
//分治操作 以x为重心的子树
void fenzhi(int x,int ssize)//ssize是当前这棵子树的大小
{
v[x]=true;
//代码保证每次进来的x都必定是当前这棵树的重心,我们将v[x]标记为true,表示x点被分治过
solve(x,0,1);//计算这棵树以x为重心的所有组合,其中包含着不成立的情况
for(int i=first[x];i;i=e[i].next)
{
int y=e[i].y;
if(v[y])continue;//分治过,不能继续运算
solve(y,e[i].z,-1);//计算不合法组合,用所有组合减去不合法组合
ms=inf;root=0;//记得要初始化
//分治的子树的大小
Size=size[y]<size[x]?size[y]:(ssize-size[x]);
getroot(y,0);//求出以y为根的子树
fenzhi(root,(size[y]<size[x]?size[y]:(ssize-size[x])));
}
}
int main()
{
scanf("%d %d",&n,&m);
for(int i=1;i<n;i++)
{
int x,y,z;
scanf("%d %d %d",&x,&y,&z);
buildroad(x,y,z);
buildroad(y,x,z);
}
for(int i=1;i<=m;i++)
scanf("%d",&ask[i]);
root=0;ms=inf;Size=n;
getroot(1,0);
fenzhi(root,n);
for(int i=1;i<=m;i++)
if(ans[i]>0)printf("AYE\n");
else printf("NAY\n");
}
A - Misha and Permutations Summation
题意:
给出两个排列,求出每个排列在全排列的排行,相加,模上n!(全排列个数)得出一个数k,求出排行为k的排列。
题解:康拓展开+树状数组+二分
按题意模拟即可
#include<bits/stdc++.h>
#define mp make_pair
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, int> pli;
typedef pair<ll, ll> pll;
typedef long double ld;
const int N=1e6+10;
const int MAXN=20010;
const int INF=0x3f3f3f3f;
const double eps=0.0000001;
const ll mod=998244353 ;
int n,m,x,y,k,cnt;
int a[N];///维护康拓系数
int c[N];///用前缀和计算小于x的数的个数
int lowbit(int x){
return x&-x;
}
int add(int x,int a)
{
while(x<=n) c[x]+=a,x+=lowbit(x);
}
int sum(int x)
{
int ans=0;
while (x) ans+=c[x],x-=lowbit(x);
return ans;
//log2n;
}
int finds(int x)
{
int l=1,r=n;
while(l < r)///寻找x的上界
{
int mid=(l+r)>>1;
int ans=sum(mid);
if(ans< x) l=mid+1;
else r=mid;
}
return l;
}
int main()
{
scanf("%d",&n);
memset(c,0,sizeof(c));
///初始化区间
for(int i=1;i<=n;i++) add(i,1);
int x;
for(int i=1;i <= n; i++)
{
scanf("%d", &x);
x++;///因为有数字0,所以整体后移一位
///找出剩余比x小的数的个数
int temp=sum(x) - 1;
///删去已经询问过的数字
add(x,-1);
a[i]+=temp;
}
for(int i=1;i<=n;i++) add(i,1);
for(int i=1;i <= n; i++)
{
scanf("%d", &x); x++;
int temp=sum(x) - 1;
add(x,-1);
a[i]+=temp;
}
for(int i=n;i>=1;i--)
{
///逆康拓展开,求余数
a[i-1]+=a[i]/(n - i + 1);
///因为上式a[i]不可能大于(n-i+1)
a[i] %=(n - i + 1);///对n!取余
}
///逆康拓展开
for(int i=1;i<=n;i++) add(i,1);
for(int i=1;i<=n;i++)
{
int tmp=finds(a[i]+1);
printf("%d ",tmp -1);
add(tmp,-1);
}
printf("\n");
}
C - Color
组合数计算公式:
题解:(容斥+组合数+逆元)
容斥有 i个颜色没有选,i个颜色没有选的概率是Ai
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#define mp make_pair
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
typedef long double ld;
const int N=1e6+10;
const int MAXN=20010;
const int INF=0x3f3f3f3f;
const double eps=0.0000001;
const LL mod=1e9+7;
LL mul(LL a,LL b){return (a%mod*b%mod)%mod;}
LL add(LL a,LL b){return (a%mod+b%mod+mod)%mod;}
LL del(LL a,LL b){return (a%mod-b%mod+mod)%mod;}
LL inv[N];
LL pre[N];
int t,k,n,m;
LL qpow(LL a,LL b,LL M){
LL r=1;
while(b){
if(b&1) r=r*a%M;
b>>=1;
a=a*a%M;
}
return r;
}
int cm[N],ck[N];
/// C(m,i) = C(m,i-1)*(m-i+1)/i;
void get_c()///计算组合数
{
cm[0]=1,ck[0]=1;
for(int i=1;i<=k;i++)
{
cm[i]=mul(mul(cm[i-1],m-i+1),inv[i]);
ck[i]=mul(mul(ck[i-1],k-i+1),inv[i]);
}
}
int main()
{
inv[1] = 1;
for(int i=2;i<=N;++i) inv[i]=qpow(i,mod-2,mod);///逆元打表
scanf("%d",&t);
LL s=t;
while(t--)
{
scanf("%d%d%d",&n,&m,&k);
get_c();
LL fla=1,ans=0;
for(int i=k;i>=1;i--)///容斥
{
ans=add(ans,mul(fla*ck[i]*i,qpow(i-1,n-1,mod)));
fla=-fla;
//cout<<ans<<" ";
}
ans=mul(ans,cm[k]);
printf("Case #%d: ",s-t);
printf("%lld\n",ans);
}
return 0;
}
D - Number Sequence
题解:(隔板法)+容斥
把 m个小球放到 n个盒子里,盒子可以为空的组合=
C
n
+
m
−
1
n
−
1
C_{n+m-1}^{n-1}
Cn+m−1n−1
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<map>
#define mp make_pair
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
typedef long double ld;
const int N=1e6+10;
const int MAXN=20010;
const int INF=0x3f3f3f3f;
const double eps=0.0000001;
const LL mod=1e9+7;
LL mul(LL a,LL b){return (a%mod*b%mod)%mod;}
LL add(LL a,LL b){return (a%mod+b%mod+mod)%mod;}
LL del(LL a,LL b){return (a%mod-b%mod+mod)%mod;}
LL C[505][505];
LL b[25] ;
map<int,int>m ;
int n ;
void init(){///杨辉三角推组合数
memset(C,0,sizeof(C));
C[0][0] = 1 ;
for( int i = 1 ; i <= 500 ; i++ ){
C[i][0] = 1 ;
for( int j = 1 ; j <= i ; j++ ){
C[i][j] = C[i-1][j] + C[i-1][j-1] ;
if( C[i][j] > mod ) C[i][j] -= mod ;
}
}
}
LL work(){
///分解质因数
for( int i =1 ;i<=n;i++ ){
for( int j=2;j*j<=b[i];j++ ){
while( b[i]%j==0){
m[j]++;
b[i]/=j ;
}
}
if(b[i]> 1) m[b[i]]++ ;
}
map<int,int>::iterator it ;
LL ans =1;///全部情况数
for( it=m.begin();it!=m.end();it++ ){
int num =it->second ;/// M_i 质因数的个数
ans = (ans*C[num+n-1][n-1])%mod ;
}
for( int i =1;i<n;i++ ){///容斥
LL val =C[n][i] ;
for( it = m.begin();it!=m.end();it++){
int num = it->second ;
val=( val*C[num+n-i-1][n-i-1])%mod ;
}
if(i&1)
ans=(ans-val+mod)%mod ;
else
ans=(ans+val)%mod ;
}
return ans ;
}
int main(){
init() ;
while( cin >> n ){
m.clear();
for( int i = 1 ; i <= n ; i++ )
cin >> b[i] ;
cout<<work()<<endl;
}
return 0;
}