P7297 [USACO21JAN] Telephone G

题目

题目

思路

比赛时乱推性质居然有60分！？UVA真水
实际上是个毒瘤分层图bfs……
按颜色分层：
先来个原图层，向所有$b_i$层的$i$点连边权为0的边
然后对于剩下的颜色层，每层每个点向左右连边权为1的边
如果当前层颜色可以到达$b_i$，本层第i个点向原图层第i个点连边权为0的边。
所有边都有向。
这样我们会发现，从原图层某个点到同层另一个点，只能通过多种颜色层横向走（每次花费1）和免费在同一号点上下摩擦（花费0）达到，符合题意。
时间复杂度NK
code:

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<deque>
using namespace std;
int head,tail,n,k,b[50005],dis[51*50000],q[101*50000];
bool o[55][55];
void add(int x,int y,int z)
{
	if (dis[y]) return;
	dis[y]=dis[x]+z;
	if (z==0) q[head]=y,head--;
	else tail++,q[tail]=y;
	return;
}
int main()
{
	scanf("%d%d",&n,&k);
	for (int i=1;i<=n;i++) scanf("%d",&b[i]);
	for (int i=1;i<=k;i++) for (int j=1;j<=k;j++)
	{
		char c;
		cin>>c;
		o[i][j]=c-'0';
	}
	dis[1]=1;
	head=50*50000,tail=50*50000+1;
	q[tail]=1;
	while (head<tail)
	{
		head++;
		int x=q[head];
		if (x<=n) add(x,b[x]*n+x,0);
		else
		{
			int cen=(x-1)/n,ge=(x-1)%n+1;
			if (ge!=1) add(x,x-1,1);
			if (ge!=n) add(x,x+1,1);
			if (o[cen][b[ge]]) add(x,ge,0);
		}
	}
	cout<<dis[n]-1;
	return 0;
}