YBTOJ荆轲刺秦&&洛谷P6474

YBTOJ荆轲刺秦&&洛谷P6474

题目传送门
思路： 搜索框架很好想，用一个五元组表示搜索状态（x，y，u1，u2，t），意为用u1次隐身，u2次瞬移，花了t时间，来到（x，y）。然后按不用魔法、使用瞬移、使用隐身的优先级分别枚举每种状态，进行bfs，然后加一些小剪枝即可AC。
难点： 这道题处理士兵视线的时候需要用到前缀和优化，否则时间复杂度将会达到O（n^4）。
考虑将区间覆盖问题用差分+前缀和实现：对于一个a^2的矩阵，将其看作a个一维区间，对于每个区间[l,r]，我们在位置l出+1，在位置r+1处-1，再求前缀和，即可实现区间[l,r]的覆盖。这样，时间复杂度就被优化成了O（n ^3）。
AC代码：

#include<bits/stdc++.h>
#define maxn 359
#define rr register
using namespace std;
int n,m,c1,c2,d,kk,edx,edy,anst,ansp,ansq;
int a[maxn][maxn],dx[8]={-1,-1,0,1,1,1,0,-1},dy[8]={0,1,1,1,0,-1,-1,-1}; 
char c[9];
bool b[maxn][maxn],vis[maxn][maxn]; 
struct node
{
	int x,y,p,q,t;
}e[maxn][maxn];
queue<node>q;
int main()
{
	scanf("%d%d%d%d%d",&n,&m,&c1,&c2,&d);
	for(rr int i=1;i<=n;++i)
	{
		for(rr int j=1;j<=m;++j)
		{
			scanf("%s",c);
			if(c[0]=='S')q.push((node){i,j,0,0,0}); 
			else if(c[0]=='T') edx=i,edy=j;
			else if(c[0]!='.')
			{
				sscanf(c,"%d",&kk); 
				b[i][j]=true;
				for(rr int k=max(1,i-kk+1);k<=min(n,i+kk-1);++k) a[k][max(1,j-kk+1+abs(i-k))]++,a[k][min(m+1,j+kk-abs(i-k))]--;
			}
		}
	}	
	for(rr int i=1;i<=n;++i) for(rr int j=1;j<=m;++j) a[i][j]+=a[i][j-1]; 
	while(q.size())
	{
		node l=q.front(); 
        q.pop();
		if(l.t>anst&&anst>0) continue;
        if(!vis[l.x][l.y])
		{ 
        	vis[l.x][l.y]=true;
        	e[l.x][l.y]=(node){l.x,l.y,l.p,l.q,l.t};
		}
		else
		{ 
			if(l.p+l.q<e[l.x][l.y].p+e[l.x][l.y].q) e[l.x][l.y].p=l.p,e[l.x][l.y].q=l.q;
			else if(l.p+l.q==e[l.x][l.y].p+e[l.x][l.y].q&&l.p<e[l.x][l.y].p) e[l.x][l.y].p=l.p,e[l.x][l.y].q=l.q;
			else continue; 
		}
        if(l.x==edx&&l.y==edy)
		{ 
    		anst=l.t;ansp=l.p;ansq=l.q;
			continue;
		}		
        if(l.q+1<=c2)
		{ 
        	for(rr int i=0;i<8;i+=2)
			{ 
        		int xx=l.x+dx[i]*d,yy=l.y+dy[i]*d;
				if(xx>=1&&xx<=n&&yy>=1&&yy<=m)
				{
					if(b[xx][yy]) continue;
					if(!a[xx][yy]) q.push((node){xx,yy,l.p,l.q+1,l.t+1});
				}
			}
		}		
        for(rr int i=0;i<8;++i)
		{ 
			int xx=l.x+dx[i],yy=l.y+dy[i];
			if(xx>=1&&xx<=n&&yy>=1&&yy<=m&&b[xx][yy]==false)
			{
				if(!a[xx][yy]) q.push((node){xx,yy,l.p,l.q,l.t+1});
			}
        }
        if(l.q+1<=c2)
		{ 
        	for(rr int i=0;i<8;i+=2)
			{
        		int xx=l.x+dx[i]*d,yy=l.y+dy[i]*d;
				if(xx>=1&&xx<=n&&yy>=1&&yy<=m)
				{
					if(b[xx][yy]) continue;
					if(a[xx][yy]&&l.p+1<=c1) q.push((node){xx,yy,l.p+1,l.q+1,l.t+1});
				}
			}
		}			
        for(rr int i=0;i<8;++i)
		{
			int xx=l.x+dx[i],yy=l.y+dy[i];
			if(xx>=1&&xx<=n&&yy>=1&&yy<=m&&b[xx][yy]==false)
			{
				if(a[xx][yy]&&l.p+1<=c1) q.push((node){xx,yy,l.p+1,l.q,l.t+1});
			}
        }	
	}
	if(anst) printf("%d %d %d\n",anst,ansp,ansq); 
	else printf("-1\n");
	return 0;
}