ZJOI 2014 星系调查(推导)

题意

https://loj.ac/problem/2201

思路

说白了就是一条路径上有 \(n\) 个二维坐标,求一条直线使得所有点到此直线的距离和最小。

设这条直线为 \(y=kx+b\) ,距离和为 \(\delta\)

\[\delta=\sum{(kx_i-y_i+b)^2\over k^2+1} \]

\[\delta={k^2\sum x_i^2+\sum y_i^2+nb^2-2k\sum x_iy_i+2kb\sum x_i-2b\sum y_i\over k^2+1} \]

由于 \(k,b\) 的取值互不影响,我们先假设 \(k\) 是一个常量,变形如下

\[\delta={nb^2+(2k\sum x_i-2\sum y_i)b+k^2\sum x_i^2-2k\sum x_iy_i+\sum y_i^2\over k^2+1} \]

发现就是一个关于 \(b\) 的二次方程,开口朝上,顶点处取最小值,即

\[b=-{2k\sum x_i-2\sum y_i\over 2n}={\sum y_i\over n}-k{\sum x_i\over n} \]

\(\displaystyle\bar x={\sum x_i\over n},\displaystyle\bar y={\sum y_i\over n}\)

得出 \(b=\bar y-k\bar x\)

代入并化简成关于 \(k\) 的式子得到

\[\delta={{(\sum x_i^2-2\bar x\sum x_i+n\bar x^2)k^2+(-2\sum x_iy_i+2\bar y \sum x_i+2\bar x\sum y_i-2n\bar x\bar y)k+(\sum y_i^2-2\bar y\sum y_i+n\bar y^2)}\over k^2+1} \]

\[\begin{array}{} A&=\sum x_i^2-2\bar x\sum x_i+n\bar x^2\\ &=\sum x_i^2-{(\sum x_i)^2\over n}\\ B&=-2\sum x_iy_i+2\bar y \sum x_i+2\bar x\sum y_i-2n\bar x\bar y\\ &=-2\sum x_iy_i+2{\sum x_i\sum y_i\over n}\\ C&=\sum y_i^2-2\bar y\sum y_i+n\bar y^2\\ &=\sum y_i^2-{(\sum y_i)^2\over n} \end{array} \]

那么

\[\delta={Ak^2+Bk+C\over k^2+1} \]

\(k\) 当作主元化简得

\[(A-\delta)k^2+Bk+C-\delta=0 \]

这个二次方程的 \(\Delta\)\(B^2-4(A-\delta)(C-\delta)\)

\[B^2-4(A-\delta)(C-\delta)\ge 0\\ -4\delta ^2+4(A+C)\delta-4AC+B^2 \ge 0 \]

解得 \(\displaystyle\delta={A+C\pm \sqrt{A^2-2AC+B^2+C^2}\over 2}\)

根号前取负号即可达到最小值.

所以维护六元组 \((\sum x_i,\sum y_i,\sum x_i^2,\sum y_i^2,\sum x_iy_i,\sum1)\) 即可求出 \(A,B,C\) ,求出 \(\delta\) 的最小值。

树的情况只用维护到根路径上的信息即可。

基环树的情况类似,断开一条环边,再枚举可行路径(至多两条)即可。

代码

#include<bits/stdc++.h>
#define FOR(i,x,y) for(int i=(x),i##END=(y);i<=i##END;++i)
#define DOR(i,x,y) for(int i=(x),i##END=(y);i>=i##END;--i)
template<typename T,typename _T>inline bool chk_min(T &x,const _T y){return y<x?x=y,1:0;}
template<typename T,typename _T>inline bool chk_max(T &x,const _T y){return x<y?x=y,1:0;}
typedef long long ll;
const int N=(int)1e5+5;
template<const int maxn,const int maxm,typename T>struct Linked_list
{
	int head[maxn],nxt[maxm],tot;T to[maxm];
	Linked_list(){clear();}
	T &operator [](const int x){return to[x];}
	void clear(){memset(head,-1,sizeof(head)),tot=0;}
	void add(int u,T v){to[++tot]=v,nxt[tot]=head[u],head[u]=tot;}
	#define EOR(i,G,u) for(int i=G.head[u];~i;i=G.nxt[i])
};
struct DisjointSet
{
	int fa[N];
	void init(int n){FOR(i,1,n)fa[i]=i;}
	int get_fa(int x){return x==fa[x]?x:fa[x]=get_fa(fa[x]);}
	void merge(int x,int y)
	{
		x=get_fa(x),y=get_fa(y);
		if(x==y)return;
		fa[x]=y;
	}
};
struct hexa
{
	int a,b,c,d,e,f;
	hexa(int _a=0,int _b=0,int _c=0,int _d=0,int _e=0,int _f=0)
	{
		a=_a,b=_b,c=_c,d=_d,e=_e,f=_f;
	}
	hexa operator +(const hexa &_)const
	{
		return hexa(a+_.a,b+_.b,c+_.c,d+_.d,e+_.e,f+_.f);
	}
	hexa operator -(const hexa &_)const
	{
		return hexa(a-_.a,b-_.b,c-_.c,d-_.d,e-_.e,f-_.f);
	}
	hexa add(int x,int y)
	{
		return hexa(a+x,b+y,c+x*x,d+y*y,e+x*y,f+1);
	}
	double solve()
	{
		double A=c-1.0*a*a/f,B=-2*e+2.0*a*b/f,C=d-1.0*b*b/f;
		return (A+C-sqrt(A*A-2*A*C+B*B+C*C))/2;
	}
};
hexa sum[N];
DisjointSet D;
Linked_list<N,N<<1,int>G;
int dep[N],fa[N],sz[N],son[N],top[N];
int X[N],Y[N];
int n,m,q;
int U,V;

void dfs(int u,int f,int d)
{
	dep[u]=d,fa[u]=f,sz[u]=1,son[u]=0;
	sum[u]=sum[f].add(X[u],Y[u]);
	EOR(i,G,u)
	{
		int v=G[i];
		if(v==f)continue;
		dfs(v,u,d+1);
		sz[u]+=sz[v];
		if(sz[v]>sz[son[u]])son[u]=v;
	}
}
void hld(int u,int f,int tp)
{
	top[u]=tp;
	if(son[u])hld(son[u],u,tp);
	EOR(i,G,u)
	{
		int v=G[i];
		if(v==f||v==son[u])continue;
		hld(v,u,v);
	}
}
int get_lca(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])std::swap(x,y);
		x=fa[top[x]];
	}
	return dep[x]<dep[y]?x:y;
}
int get_dis(int x,int y)
{
	return dep[x]+dep[y]-2*dep[get_lca(x,y)];
}
hexa get_val(int x,int y)
{
	int lca=get_lca(x,y);
	return (sum[x]-sum[lca]+sum[y]-sum[lca]).add(X[lca],Y[lca]);
}
bool intersect(int a,int b,int c,int d)
{
	int lca1=get_lca(a,b),lca2=get_lca(c,d);
	return 	get_dis(a,lca2)+get_dis(lca2,b)==get_dis(a,b)||
			get_dis(c,lca1)+get_dis(lca1,d)==get_dis(c,d);
}

int main()
{
	scanf("%d%d",&n,&m);
	FOR(i,1,n)scanf("%d%d",&X[i],&Y[i]);
	D.init(n);
	FOR(i,1,m)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		if(D.get_fa(u)==D.get_fa(v))U=u,V=v;
		else
		{
			G.add(u,v),G.add(v,u);
			D.merge(u,v);
		}
	}
	dfs(1,0,1);
	hld(1,0,1);
	scanf("%d",&q);
	while(q--)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		if(n==m-1)printf("%.5lf\n",get_val(x,y).solve());
		else
		{
			double res=get_val(x,y).solve();
			if(!intersect(x,U,y,V))chk_min(res,(get_val(x,U)+get_val(y,V)).solve());
			if(!intersect(x,V,y,U))chk_min(res,(get_val(x,V)+get_val(y,U)).solve());
			printf("%.5lf\n",res);
		}
	}
	return 0;
}
posted @ 2019-04-27 08:17  Paulliant  阅读(113)  评论(0编辑  收藏