hdu 4812 D Tree（点分治）

传送门

题意：询问一棵树上是否存在点对满足这两个点的最短路径上所有点的点权之积mod M之后等于K。若存在，则输出满足条件的字典序最小的一对点。已知M=1e6+3。

题解：一般询问树上是否存在某种“合法”路径，可以考虑点分治。如果一对点(u,v)满足，设当前重心为p，链[u,p)的点权之积为a，链[v,p)的点权之积为b，则，则。对于每个重心，相当于用某棵子树中的点与其他子树中的点匹配，用cov数组记录一下可以匹配到的点的最小编号即可。（因为M不大，所以逆元可以递推存到inv数组）

注意：

1. b*val[p]可能可以被M整除，这种情况不能作为答案（因为a*b*val[p]也会被M整除），特判一下。

2. 每次复原cov数组时不能用memset，否则会TLE，要开个rec数组单独记录一下cov数组哪些下标被修改过，考虑新的重心时把这些下标的cov值复原为INF即可

3. 一定要记得考虑从重心本身出发的链！所有点分治的题都要注意这种情况！

 

成功地A了一道之前看一眼就不会去想的题......

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll M=1e6+3;
const int N=1e5+4;
const int INF=0x3f3f3f3f;
int n,K;
int head[N],etot;
struct Edge {
	int v,nxt;
}e[N<<1];
int siz[N],mx[N];
bool vis[N];
int root,s,sum;
ll dis[N],inv[M],ctr;
int val[N],cov[M];
int x,y;
int q[N],cnt;
inline void adde(int v,int u) {
	e[++etot].nxt=head[u],e[etot].v=v,head[u]=etot;
}
inline void init() {
	x=INF,y=INF;
	etot=0;
	memset(head,-1,sizeof(head));
	memset(vis,false,sizeof(vis));
}
inline void update(int &x,int &y,int t1,int t2) {
	if (t1>t2) t1^=t2^=t1^=t2;
	if (x>t1) x=t1,y=t2;
	else if (x==t1&&y>t2) y=t2;
}
inline void smin(int &a,int b) {
	a=a<b?a:b;
}
inline void smax(int &a,int b) {
	a=a>b?a:b;
}
inline void getroot(int p,int fa) {
	mx[p]=-INF,siz[p]=1;
	for (int i=head[p];~i;i=e[i].nxt) {
		int v=e[i].v;
		if (vis[v]||v==fa) continue;
		getroot(v,p);
		siz[p]+=siz[v];
		smax(mx[p],siz[v]);
	}
	smax(mx[p],sum-siz[p]);
	if (mx[p]<s) s=mx[p],root=p;
}
inline void getdis(int p,int fa) {
	q[++cnt]=p;
	for (int i=head[p];~i;i=e[i].nxt) {
		int v=e[i].v;
		if (vis[v]||v==fa) continue;
		dis[v]=dis[p]*val[v]%M;
		getdis(v,p);
	}
}
int rec[N],tot;
inline void calc(int p,ll d) {//a*b*ctr%M=K, b=inv[a*ctr]*K%M
	cnt=0;
	dis[p]=d;
	getdis(p,0);
	for (int i=1;i<=cnt;++i) {
		int id=dis[q[i]]%M;
		if (cov[id]^INF)
			update(x,y,cov[id],q[i]);
	}
	for (int i=1;i<=cnt;++i) {
		int id=dis[q[i]]*ctr%M;
		if (id) {
			int idx=inv[id]*K%M;
			smin(cov[rec[++tot]=idx],q[i]);
		}
	}
}
inline void work(int p) {
	ctr=val[p];
	vis[p]=true;
	for (int i=1;i<=tot;++i) cov[rec[i]]=INF;
	tot=0;
	if (ctr%M) {//Consider chains that start from p itself!
		int idx=inv[ctr%M]*K%M;
		smin(cov[rec[++tot]=idx],p);
	}
	for (int i=head[p];~i;i=e[i].nxt) {
		int v=e[i].v;
		if (vis[v]) continue;
		calc(v,val[v]);
	}
	for (int i=head[p];~i;i=e[i].nxt) {
		int v=e[i].v;
		if (vis[v]) continue;
		s=INF,sum=siz[v];
		getroot(v,0);
		work(root);
	}
}
inline void getinv() {
	inv[1]=1;
	for (register int i=2;i<M;++i)
		inv[i]=(M-M/i)*inv[M%i]%M;
}
int main() {
//	freopen("in.txt","r",stdin);
	getinv();
	memset(cov,INF,sizeof(cov));
	while (~scanf("%d%d",&n,&K)) {
		init();
		for (register int i=1;i<=n;++i) scanf("%lld",&val[i]);
		for (register int i=1;i<n;++i) {
			int u,v;
			scanf("%d%d",&u,&v);
			adde(u,v);
			adde(v,u);
		}
		s=INF,sum=n;
		getroot(1,0);
		work(root);
		if (x^INF) printf("%d %d\n",x,y);
		else puts("No solution");
	}
	return 0;
}