【THUSC2017】【LOJ2982】宇宙广播 计算几何 高斯消元

题目大意

  有 \(n\)\(n\) 维空间中的球,求这些球的所有公切面。

  保证不会无解或有无穷多组解。

  \(n\leq 10\)

题解

  你可以认为这是一道传统题。

  记公切面为 \(a_1x_1+a_2x_2+\cdots+ a_nx_n=d\),满足 \(\sum_ia_i^2=1\)

  一个点 \(x_1,x_2,\ldots,x_n\) 到这个面的距离为

\[\frac{\lvert a_1x_1+a_2x_2\cdots +a_nx_n-d\rvert}{\sqrt{a_1^2+a_2^2+\cdots+a_n^2}} \]

  对于第 \(i\) 个球,有 \(\lvert\sum_{j}a_jx_{i,j}-d\rvert=r\)

  枚举所有 \(\lvert\sum_{j}a_jx_{i,j}-d\rvert\) 的符号,把绝对值符号去掉。

  然后解方程解出 \(a_i=D_1d+D_2\),带进 \(\sum_ia_i^2=1\) 中解方程就可以解出 \(d\)

  时间复杂度:\(O(n^32^n)\)

  直接解方程可能会有奇怪情况,所以要把所有坐标偏移一个值。

代码

const ldb eps=1e-6;
ldb DELTA[]={0,0.235,0.3746,0.1246,0.2153,0.364,0.1253,0.6124,0.164,0.7543,0.35476};
int n;
ldb a[20][20];
ldb r[20];
ldb ans[2100][20][20];
ldb b[20][20];
ldb c[20][20];
ldb e[20];
int cnt;
void gao(ldb d)
{
	for(int i=1;i<=n;i++)
		e[i]=c[i][n+1]*d+c[i][n+2];
	cnt++;
	for(int i=1;i<=n;i++)
	{
		ldb s=0;
		for(int j=1;j<=n;j++)
			s+=a[i][j]*e[j];
		s-=d;
		for(int j=1;j<=n;j++)
			ans[cnt][i][j]=a[i][j]-s*e[j];
	}
}
void calc()
{
	memcpy(c,b,sizeof b);
	int flag=0;
	for(int i=1;i<=n;i++)
	{
		int x=i;
		for(int j=i;j<=n;j++)
			if(fabs(c[j][i])>fabs(c[x][i]))
				x=j;
		if(fabs(c[x][i])<eps)
		{
			flag=1;
			continue;
		}
		if(x!=i)
			swap(c[x],c[i]);
		ldb v=1/c[i][i];
		for(int j=1;j<=n+2;j++)
			c[i][j]*=v;
		for(int j=1;j<=n;j++)
			if(j!=i)
			{
				ldb v=c[j][i];
				for(int k=1;k<=n+2;k++)
					c[j][k]-=c[i][k]*v;
			}
	}
	if(flag)
	{
//		ldb d;
//		for(int i=1;i<=n;i++)
//			if(!c[i][i])
//			{
//				if(flag)
//				{
//				}
//				else
//				{
//				}
		printf("error\n");
		return;
	}
	ldb A=0,B=0,C=0;
	for(int i=1;i<=n;i++)
	{
		A+=c[i][n+1]*c[i][n+1];
		B+=2*c[i][n+1]*c[i][n+2];
		C+=c[i][n+2]*c[i][n+2];
	}
	C-=1;
	if(fabs(A)<eps)
	{
		if(fabs(B)<eps)
		{
			return;
		}
		ldb d=-C/B;
		gao(d);
	}
	else
	{
		ldb delta=B*B-4*A*C;
		if(delta<-eps)
			return;
		else
			delta=max(delta,(ldb)0);
		if(delta<eps)
		{
			ldb d=-B/(2*A);
			gao(d);
		}
		else
		{
			ldb d1=(-B+sqrt(delta))/(2*A);
			gao(d1);
			ldb d2=(-B+sqrt(delta))/(2*A);
			gao(d2);
		}
	}
}
void dfs(int x)
{
	if(x>n)
	{
		calc();
		return;
	}
	for(int i=1;i<=n;i++)
		b[x][i]=a[x][i];
	b[x][n+1]=1;
	b[x][n+2]=r[x];
	dfs(x+1);
	if(fabs(r[x])<eps)
		return;
	b[x][n+2]=-r[x];
	dfs(x+1);
}
void solve()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			scanf("%Lf",&a[i][j]);
			a[i][j]-=DELTA[j];
		}
		scanf("%Lf",&r[i]);
	}
	cnt=0;
	dfs(1);
	printf("%d\n",cnt);
	for(int i=1;i<=cnt;i++)
		for(int j=1;j<=n;j++)
		{
			for(int k=1;k<=n;k++)
				printf("%.20Lf ",ans[i][j][k]+DELTA[k]);
			printf("\n");
		}
}
int main()
{
//	freopen("0.in","r",stdin);
	int t;
	scanf("%d",&t);
	while(t--)
		solve();
	return 0;
}
posted @ 2019-01-20 13:45  ywwyww  阅读(...)  评论(...编辑  收藏