暑假集训日记——8.4( 点分治+康拓展开+组合数)

点分治详解:大佬的博客
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+m1n1

#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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值