题目:

我是超链接

题意:

一棵树,每条边上有一个数字(1~9),给出一个与10互质的数m,问整棵树上有多少条链满足从起点走到终点树链上形成的十进制数是m的倍数。

题解:

处理树上路径,点分治吧
依然考虑lca为分治中心的情况,我们需要维护三个量,从下往上走在%m意义下的值,从上往下走在%m意义下的值,每个点的深度(便于将其合起来)
那么一条合格的路径应该是这样的 up(x)10deep(y)+down(y)0(%m) u p ( x ) ∗ 10 d e e p ( y ) + d o w n ( y ) ≡ 0 ( % m )
移项就可以得到 axb(%m) a x ≡ b ( % m ) 这样的形式,那我们对于每个down(y)寻找合适的up(x)就好了,up(x)可以通过exgcd计算出来
怎么说细节还是挺多,加longlong%m什么的,最重要的是,这里不能保证树高,mi数组要开到n!

代码:

#include <cstdio>
#include <iostream>
#include <algorithm>
#define LL long long
#define INF 1e9
using namespace std;
const int N=100005;
int m,n,tot,nxt[N*2],point[N],v[N*2],c[N*2],size[N],f[N],sum,num,root,d[N],deep[N];bool vis[N];
LL up[N],down[N],ans,mi[N],zup[N];
void addline(int x,int y,int z)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
int gcd(int a,int b){if (!b) return a;else return gcd(b,a%b);}
void exgcd(int a,int b,int &x,int &y){if (!b) x=1,y=0;else exgcd(b,a%b,y,x),y-=a/b*x;}
void find(int x,int fa)
{
    size[x]=1; f[x]=0;
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa && !vis[v[i]])
      {
        find(v[i],x);
        size[x]+=size[v[i]];
        f[x]=max(f[x],size[v[i]]);
      }
    f[x]=max(f[x],sum-size[x]);
    if (f[x]<f[root]) root=x;
}
void dfs(int x,int fa,int dep)
{
    d[++num]=x; up[num]=zup[x]; deep[x]=dep;
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa && !vis[v[i]])
      {
        zup[v[i]]=(mi[dep]*(LL)c[i]%m+zup[x])%m;
        down[v[i]]=(10LL%m*down[x]%m+c[i])%m;
        dfs(v[i],x,dep+1);
      }
}
LL calc(int x,int nowlen)
{
    zup[x]=down[x]=nowlen%m; num=0;
    dfs(x,0,(nowlen)?1:0);LL ans=0;int mm=m;
    sort(up+1,up+num+1);
    for (int i=1;i<=num;i++)
    {
        m=mm;
        int b=(m-down[d[i]]%m)%m,x,y;
        int a=mi[deep[d[i]]],t=gcd(a,m);
        if (b%t) continue;
        a/=t; m/=t; b/=t;
        exgcd(a,m,x,y);
        x=(x%m*(LL)b%m+m)%m;
        int l=lower_bound(up+1,up+num+1,x)-up;
        if (up[l]!=x) continue;
        int r=upper_bound(up+1,up+num+1,x)-up-1; 
        ans+=(LL)(r-l+1);
    }
    m=mm;
    if (!nowlen) ans--;
    return ans;
}
void work(int x)
{
    ans+=calc(x,0); vis[x]=1;
    for (int i=point[x];i;i=nxt[i])
      if (!vis[v[i]])
      {
        ans-=calc(v[i],c[i]);
        f[0]=INF; root=0; sum=size[v[i]]; find(v[i],x);
        work(root);
      }
}
int main()
{
    scanf("%d%d",&n,&m);
    mi[0]=1; for (int i=1;i<=n;i++) mi[i]=(LL)mi[i-1]*10%m;
    for (int i=1;i<n;i++)
    {
        int x,y,z;scanf("%d%d%d",&x,&y,&z);
        x++;y++;addline(x,y,z);
    }
    f[0]=INF; root=0; sum=n; find(1,0); 
    work(root);
    printf("%I64d",ans);
}
Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐