【NOIP2017】棋盘

题目描述
有一个m × m的棋盘,棋盘上每一个格子可能是红色、黄色或没有任何颜色的。你现在
要从棋盘的最左上角走到棋盘的最右下角。
任何一个时刻,你所站在的位置必须是有颜色的(不能是无色的),你只能向上、下、
左、右四个方向前进。当你从一个格子走向另一个格子时,如果两个格子的颜色相同,那你
不需要花费金币;如果不同,则你需要花费1 个金币。
另外,你可以花费2 个金币施展魔法让下一个无色格子暂时变为你指定的颜色。但这个
魔法不能连续使用,而且这个魔法的持续时间很短,也就是说,如果你使用了这个魔法,走
到了这个暂时有颜色的格子上,你就不能继续使用魔法;只有当你离开这个位置,走到一个
本来就有颜色的格子上的时候,你才能继续使用这个魔法,而当你离开了这个位置(施展魔
法使得变为有颜色的格子)时,这个格子恢复为无色。
现在你要从棋盘的最左上角,走到棋盘的最右下角,求花费的最少金币是多少?

输入
第一行包含两个正整数m,n,以一个空格分开,分别代表棋盘的大小,棋盘上
有颜色的格子的数量。
接下来的 n 行,每行三个正整数x,y,c,分别表示坐标为(x,y)的格子有颜色c。
其中c=1 代表黄色,c=0 代表红色。相邻两个数之间用一个空格隔开。棋盘左上角的坐标
为(1, 1),右下角的坐标为(m, m)。

棋盘上其余的格子都是无色。保证棋盘的左上角,也就是(1,1)一定是有颜色的。

输出
输出一行,一个整数,表示花费的金币的最小值,如果无法到达,输出-1。

样例输入

5 7
1 1 0
1 2 0
2 2 1
3 3 1
3 4 0
4 4 1
5 5 0

样例输出

8

非常明显,这道题可是dfs啊…
但是,看看这个数据范围题目要求普通的dfs肯定超时呀,这题,考试的时候吃了个大亏,想了一个多小时,才写了这么个dfs…

#include
#include
#define in(q,p)(q
const int N=100,wa[][2]={{0,1},{0,-1},{1,0},{-1,0}};
int n,num,x,y,col,ans=2e8;
int ma[N+5][N+5];
bool fg[N+5][N+5],can;
int readint()
{
	int x=0;char c=getchar();
	while(c<'0'||c>'9') c=getchar();
	while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
	return x;
}
void dfs(int nx,int ny,int t,bool f)
{
	fg[nx][ny]=1;
	if(nx==n&&ny==n)
	{
		ans=in(ans,t);
		return ;
	}
	int x1,y1;
	for(int i=0;i<4;i++)
	{
		x1=nx+wa[i][0];
		y1=ny+wa[i][1];
		if(x1<1||x1>n||y1<1||y1>n) continue;
		if(ma[x1][y1]!=-1)
		{
			if(fg[x1][y1]==-1||!fg[x1][y1])
			{
				dfs(x1,y1,t+(ma[x1][y1]!=ma[nx][ny]),0);
				fg[x1][y1]=0;
			}
		}
		if(ma[x1][y1]==-1&&!f)
		{
			if(fg[x1][y1]==-1||!fg[x1][y1])
			{
				ma[x1][y1]=ma[nx][ny];
				dfs(x1,y1,t+2,1);
				fg[x1][y1]=0;
			}
		}
	}
	if(f)  ma[nx][ny]=-1;
}
int main()
{
	//freopen("chess.in","r",stdin);
	//freopen("chess.out","w",stdout);
	n=readint();num=readint();
	memset(ma,-1,sizeof(ma));
	memset(fg,-1,sizeof(fg));
	for(int i=1;i<=num;i++)
	{
		x=readint();
		y=readint();
		col=readint();
		ma[x][y]=col;
	}
	dfs(1,1,0,0);
	if(ans!=2e8) printf("%d",ans);
	else printf("-1");
	return 0;
}

反正测下来就20分…
自己在洛谷上测35,在清北学堂的数据上测也是35…反正noi的数据就是这么坑!

不说这么多,听同学说,要剪枝,于是改出来了这么个代码:

#include
#include
const int N=100,wa[][2]={{0,1},{1,0},{-1,0},{0,-1}};
int n,num,x,y,col,bol;
int ma[N+5][N+5],fg[N+5][N+5];
bool can;
inline int readint()
{
    int x=0;char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x;
}
inline void dfs(int nx,int ny,int t,bool f)
{
    if(fg[nx][ny]!=-1&&t>=fg[nx][ny]) return ;
    fg[nx][ny]=t;
    if(nx==n&&ny==n)
        return ;
    int x1,y1;
    for(int i=0;i<4;i++)
    {
        x1=nx+wa[i][0];
        y1=ny+wa[i][1];
        if(x1<1||x1>n||y1<1||y1>n) continue;
        if(ma[x1][y1]!=-1)
        {
            bol=(ma[x1][y1]!=ma[nx][ny]);
            dfs(x1,y1,t+bol,0);
        }
        if(ma[x1][y1]==-1&&!f)
        {
            ma[x1][y1]=ma[nx][ny];
            dfs(x1,y1,t+2,1);
            ma[x1][y1]=-1;
        }
    }
}
int main()
{
    n=readint();num=readint();
    memset(ma,-1,sizeof(ma));
    memset(fg,-1,sizeof(fg));
    for(int i=1;i<=num;i++)
    {
        x=readint();
        y=readint();
        col=readint();
        ma[x][y]=col;
    }
    dfs(1,1,0,0);
    printf("%d",fg[n][n]);
    return 0;
}

然后交过了。

正文结束


一个小插曲(思考题):

在写剪枝的时候,我之前是写的:

inline void dfs(int nx,int ny,int t,bool f)
{
    if(fg[nx][ny]!=-1&&t>=fg[nx][ny]) return ;
    fg[nx][ny]=t;
    if(nx==n&&ny==n)
        return ;
    int x1,y1;
    for(int i=0;i<4;i++)
    {
        x1=nx+wa[i][0];
        y1=ny+wa[i][1];
        if(x1<1||x1>n||y1<1||y1>n) continue;
        if(ma[x1][y1]!=-1)
        {
            bol=(ma[x1][y1]!=ma[nx][ny]);
            dfs(x1,y1,t+bol,0);
        }
        if(ma[x1][y1]==-1&&!f)
        {
            ma[x1][y1]=ma[nx][ny];
            dfs(x1,y1,t+2,1);
        }
    }
    if(f) ma[nx][ny]=-1;
}

然后总是有两三组数据没有过,都是少了个2…
没错,就是这一行:

if(f) ma[nx][ny]=-1;

不知道为什么会错,可能noi上面的就是这种数据给卡掉了几组吧…

反正改成这样就对了呗:

inline void dfs(int nx,int ny,int t,bool f)
{
    if(fg[nx][ny]!=-1&&t>=fg[nx][ny]) return ;
    fg[nx][ny]=t;
    if(nx==n&&ny==n)
        return ;
    int x1,y1;
    for(int i=0;i<4;i++)
    {
        x1=nx+wa[i][0];
        y1=ny+wa[i][1];
        if(x1<1||x1>n||y1<1||y1>n) continue;
        if(ma[x1][y1]!=-1)
        {
            bol=(ma[x1][y1]!=ma[nx][ny]);
            dfs(x1,y1,t+bol,0);
        }
        if(ma[x1][y1]==-1&&!f)
        {
            ma[x1][y1]=ma[nx][ny];
            dfs(x1,y1,t+2,1);
            ma[x1][y1]=-1;
        }
    }
}

虽然我还是不知道为什么…

你可能感兴趣的:(回溯,noip-2017)