标签：head mxson int ll codeforces 逆元 x2 x1 715C

codeforces 715C(点分治+逆元

题意：

给一个n个点的树，问多少条路径满足边权连成的数%M==0

思路：

考虑点分治按容斥的写法计数，这题细节很多。

首先x1表示从根下来的路径，x2表示上去的路径，维护就很简单了

x 1 ∗ 1 0 ( d e p 2 ) + x 2 = 0 ( m o d m ) x1*10^{(dep2)}+x2=0(modm) x1∗10(dep2)+x2=0(modm)

我们维护将x2加进桶里面，查询前面的数即可，这里有一些细节注意的是，我们得考虑端点，在分治重心的位置的时候，统计重心为端点的两条路径，具体实现标记一下，注意这种路径在容斥的时候是不用再算的，最后不得不吐槽一下，出题人数据居然卡了unordered_map的hash，服了

map能过，unordered_map不能过

#include<bits/stdc++.h>
#define fi first 
#define se second 
using namespace std;
typedef long long ll;
typedef pair<ll,int> P;
const int maxn=1e5+5;
map<ll,ll>mp;
int n,m,head[maxn],ver[maxn<<1],next1[maxn<<1],edge[maxn<<1],tot,sz[maxn],mxson[maxn],rt,S,cnttmp;
ll ans=0,p[maxn];
P tmp[maxn];
bool v[maxn];
inline int read(){
    char c=getchar();int x=0;
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
    return x;
}
inline void in(int&x){
    x=read();
}
void add(int x,int y,int z){
    ver[++tot]=y,next1[tot]=head[x],edge[tot]=z,head[x]=tot;
}
void exgcd(ll a,ll b,ll&x,ll&y){
    if(!b){x=1;y=0;return;}
    exgcd(b,a%b,x,y);
    ll z=x;x=y;y=z-y*(a/b);
}
ll inv(ll a){
    ll x,y;
    exgcd(a,m,x,y);
    return (x%m+m)%m;
}
void getRoot(int x,int f){
    sz[x]=1;mxson[x]=0;
    for(int i=head[x];i;i=next1[i]){
        int y=ver[i];
        if(y==f||v[y])continue;
        getRoot(y,x);
        sz[x]+=sz[y];
        mxson[x]=max(mxson[x],sz[y]);
    }
    mxson[x]=max(mxson[x],S-mxson[x]);
    if(mxson[x]<mxson[rt])rt=x;
}
void init(int x){
    mxson[rt=0]=maxn;
    S=sz[x];
    getRoot(x,0);
}
void getQue(int x,int f,ll x1,ll x2,int d){
    if(d>=0)++mp[x2],tmp[++cnttmp]={x1,d};
    for(int i=head[x];i;i=next1[i]){
        int y=ver[i];
        if(v[y]||y==f)continue;
        getQue(y,x,(x1*10+edge[i])%m,(x2+1ll*edge[i]*p[d+1])%m,d+1);
    }
}
ll cal(int x,ll val){
    mp.clear();
    cnttmp=0;
    ll sum=0;
    if(!val)getQue(x,0,0,0,-1);//单个点不算所以不统计(u,u)
    else getQue(x,0,val%m,val%m,0);   
    for(int i=1;i<=cnttmp;++i){
        ll flag=(-tmp[i].fi*inv(p[tmp[i].se+1])%m+m)%m;
        sum+=mp[flag];
        if(!val&&!tmp[i].fi)sum++;//自顶向下的单条链%m==0
    }
    if(!val)sum+=mp[0];//自底向上单条链%m==0
    return sum;
}
void dfz(int x){
    v[x]=1;
    ans+=cal(x,0);
    for(int i=head[x];i;i=next1[i]){
        int y=ver[i];
        if(v[y])continue;
        ans-=cal(y,edge[i]);
        init(y);
        dfz(rt);
    }
    v[x]=0;
}

int main(){
    in(n);in(m);
    int a,b,c;
    for(int i=1;i<n;++i){
        in(a);in(b);in(c);
        a++;b++;
        add(a,b,c);add(b,a,c);
    }
    p[0]=1;
    for(int i=1;i<=n;++i){
        p[i]=p[i-1]*10%m;
    }
    mxson[rt=0]=maxn;
    S=n;
    getRoot(1,0);
    dfz(rt);
    cout<<ans<<"\n";
    return 0;
}

标签：head,mxson,int,ll,codeforces,逆元,x2,x1,715C
来源： https://blog.csdn.net/weixin_45539557/article/details/115716690

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