【CF#715C】Digit Tree 点分治+乘法逆元

AC通道:http://codeforces.com/problemset/problem/715/C

【题目大意】

给定一个有N个点的树,问其中有多少条路径满足他们的边权连成的数对M取余为0。

其中gcd(M,10)=1。

【题解】

首先是点分治的套路,然后这题的主要难点在如何统计连通块中的答案。

对于x→y的路径,也就是x→root→y,我们可以处理出dis[x]和dis[y]分别表示x到root、root到y连成的数对mod取模的结果

那么在合并时,如果路径合法,则应满足以下条件:


即:

然后我们就可以愉快地使用dfs统计答案了。

#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
#define FILE "read"
#define MAXN 1000010
#define INF 1000000000
#define up(i,j,n) for(ll i=j;i<=n;++i)
#define dn(i,j,n) for(ll i=j;i>=n;--i)
#define cmax(a,b) a=max(a,b)
#define cmin(a,b) a=min(a,b)
namespace INIT{
	char buf[1<<15],*fs,*ft;
	inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline ll read(){
		ll x=0,f=1;  char ch=getc();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
		return x*f;
	}
}using namespace INIT;
struct node{ll y,next,v;}e[MAXN*2];
ll n,mod,sum,root,len,ans,Link[MAXN],dis[MAXN],inv[MAXN],p[MAXN],deep[MAXN],size[MAXN],f[MAXN],vis[MAXN];
mapcnt;
void insert(ll x,ll y,ll v){e[++len].next=Link[x];Link[x]=len;e[len].y=y;e[len].v=v;}
void exgcd(ll a,ll b,ll &x,ll &y){
	if(!b) {x=1;y=0;return;}
	exgcd(b,a%b,x,y);
	ll t=x;x=y;y=t-a/b*y;
}
ll Ni(ll num){
	ll x,y; exgcd(num,mod,x,y);
	return (x%mod+mod)%mod;
}
void pre(){
	p[0]=1;up(i,1,n)p[i]=p[i-1]*10%mod;
	up(i,0,n)inv[i]=Ni(p[i]);
}
void getroot(ll x,ll fa){
	f[x]=0; size[x]=1;
	for(ll i=Link[x];i;i=e[i].next){
		if(e[i].y==fa||vis[e[i].y])  continue;
		getroot(e[i].y,x);  size[x]+=size[e[i].y];
		cmax(f[x],size[e[i].y]);
	}
	cmax(f[x],sum-size[x]);
	if(f[x]


你可能感兴趣的:(codeforces,乘法逆元,点分治)