「SCOI 2018」Numazu 的蜜柑

forever_dreams

于 2019-11-10 22:54:08 发布

阅读量332

点赞数

分类专栏： --------数学问题-------- # 二次剩余文章标签：二次剩余

本文链接：https://blog.csdn.net/forever_dreams/article/details/102980270

版权

--------数学问题-------- 同时被 2 个专栏收录

141 篇文章 1 订阅

订阅专栏

二次剩余

2 篇文章 0 订阅

订阅专栏

传送门

problem

给定一个 $n$ 的点的有根树（根为 $1$ ），点 $u$ 有权值 $a_u$ 。统计满足下列条件的点对 $(u, v)$ 的数量：

$v$ 是 $u$ 的父亲。
$a_u^{\,2}+Aa_ua_v+Ba_v^{\,2}\equiv 0\pmod p$

其中 $A, B, p$ 都是给定的数。

数据范围： $n\le 10^5$ ， $3\le p\le10^{16}$ 。

solution

这好像是二次剩余的模板题啊。

首先可以通过求根公式直接解出那个式子： $a_u=\frac{-A\pm\sqrt{A^2-4B}}{2}a_v$ 。

那么讨论一下 $\sqrt{A^2-4B}$ 是否有解：

若 $\sqrt{A^2-4B}$ 无解。那么就相当于统计 $a_u=a_v=0$ 的答案。
若 $\sqrt{A^2-4B}$ 有解。就按照上式（即 $a_u=\frac{-A\pm\sqrt{A^2-4B}}{2}a_v$ ）统计答案。

由于 $v$ 是 $u$ 的祖先，这就非常好做，用个 map 存一下值即可。

时间复杂度 $O (n)$ 。

code

#include<bits/stdc++.h>
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
namespace IO{
	const int Rlen=1<<22|1;
	char buf[Rlen],*p1,*p2;
	inline char gc(){
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	template<typename T>
	inline T Read(){
		char c=gc();T x=0,f=1;
		while(!isdigit(c))  f^=(c=='-'),c=gc();
		while( isdigit(c))  x=(x+(x<<2)<<1)+(c^48),c=gc();
		return f?x:-x;
	}
	inline int gi()  {return Read<int>();}
	inline ll  gl()  {return Read<ll >();}
}
using IO::gi;
using IO::gl;
const int N=1e5+5;
ll A,B,P,inv2,x1,x2,sum,a[N];
inline ll add(ll x,ll y)  {return x+y>=P?x+y-P:x+y;}
inline ll dec(ll x,ll y)  {return x-y< 0?x-y+P:x-y;}
inline ll mul(ll x,ll y)  {return (x*y-(ll)((long double)x/P*y)*P+P)%P;}
inline ll power(ll a,ll b){
	ll ans=1;
	for(;b;b>>=1,a=mul(a,a))  if(b&1)  ans=mul(ans,a);
	return ans;
}
inline ll Inv(ll x)  {return power(x,P-2);}
int n,t,first[N],v[N],nxt[N];
inline void edge(int x,int y)  {nxt[++t]=first[x],first[x]=t,v[t]=y;}
namespace Cipolla{
	ll I,val;
	struct num{
		ll x,y;
		num(ll x=0,ll y=0):x(x),y(y){}
		inline friend num operator*(const num &a,const num &b){
			return num(add(mul(a.x,b.x),mul(val,mul(a.y,b.y))),add(mul(a.x,b.y),mul(a.y,b.x)));
		}
		inline friend num operator^(num a,ll b){
			num ans(1,0);
			for(;b;b>>=1,a=a*a)  if(b&1)  ans=ans*a;
			return ans;
		}
	};
	inline pll Sqrt(ll n){
		if(!n)  return pll(0,0);
		if(power(n,(P-1)>>1)==P-1)  return pll(-1,-1);
		while(1){
			I=rand(),val=dec(mul(I,I),n);
			if(power(val,(P-1)>>1)==P-1)  break;
		}
		ll x=(num(I,1)^((P+1)>>1)).x;
		return pll(x,P-x);
	}
}
using Cipolla::Sqrt;
int cnt=0;
inline void dfs0(int x){
	if(!a[x])  sum+=cnt,cnt++;
	for(int i=first[x];i;i=nxt[i])  dfs0(v[i]);
	if(!a[x])  cnt--;
}
unordered_map<ll,int>num;
inline void dfs1(int x){
	sum+=num[a[x]];
	ll val1=mul(a[x],x1),val2=mul(a[x],x2);
	(val1==val2)?(++num[val1]):(++num[val1],++num[val2]);
	for(int i=first[x];i;i=nxt[i])  dfs1(v[i]);
	(val1==val2)?(--num[val1]):(--num[val1],--num[val2]);
}
int main(){
	n=gi(),P=gl(),A=gl()%P,B=gl()%P;
	pll delta=Sqrt(dec(mul(A,A),mul(4,B)));
	inv2=Inv(2),x1=mul(add(P-A,delta.fi),inv2),x2=mul(add(P-A,delta.se),inv2);
	for(int i=1;i<=n;++i)  a[i]=gl();
	for(int i=2,fa;i<=n;++i)  fa=gi(),edge(fa,i);
	(delta.fi==-1)?(dfs0(1),1):(dfs1(1),1);
	printf("%lld\n",sum);
	return 0;
}