中石油问题 B: Kinky Word Search(搜索减枝)

题目链接

可能就不去~~~

题意:

给出你一个地图,和拐弯次数,给出你一个路径,看你能在规定的拐弯次数走路径一边.

思路:

因为数据量比较小,我们可以想到直接用dfs来搜,就是沿着路径走,中间记录走的步数,拐弯次数,和原本方向.当然我们需要减枝,可以直接用数据减枝
(10101008100)的数组还是可以开出来的,走过就说明相同状态,不用考虑.
注意 当1*1的地图,路径如果是1,而拐弯次数是0,那也可以实现

不减枝应该会T80多,不特判最后一个应该会WA97

#include <set>
#include <map>
#include <queue>
#include <string>
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
typedef pair<ll, ll> pii;
#define mem(a,x) memset(a,x,sizeof(a))
#define debug(x) cout << #x << ": " << x << endl;
#define rep(i,n) for(int i=0;i<(n);++i)
#define repi(i,a,b) for(int i=int(a);i<=(b);++i)
#define repr(i,b,a) for(int i=int(b);i>=(a);--i)
const int maxn = 2e5 + 1010;
#define inf 0x3f3f3f3f
#define x first
#define y second
#define sf scanf
#define pf printf
const int mod = 998244353;
const int MOD = 10007;
const int dx[] = {
    
    1, 1, 1, 0, 0, -1, -1, -1};
const int dy[] = {
    
    1, 0, -1, -1, 1, -1, 1, 0};
 
ll a[maxn], b[maxn];
ll ans, n, m, len, nn;
string s;
#define read read()
bool d[11][11][101][8][101];
char dp[200][200];
 
void dfs(ll x, ll y, ll sum, ll flag, ll num)
{
    
    
    if(num == len && sum == m)
    {
    
    
        ans = 1;
        return ;
    }
    if(num >= len) return ;
    if(sum > m) return ;
 
    if(sum >= 0 && flag >= 0 && d[x][y][sum][flag][num])
    {
    
    
        return ;
    }
    else if(sum >= 0 && flag >= 0 && !d[x][y][sum][flag][num])
    {
    
    
        d[x][y][sum][flag][num] = 1;
    }
 
    for(int i = 0; i < 8; i++)
    {
    
    
        ll fx = dx[i] + x, fy = dy[i] + y;
        if(fx <= n && fx >= 1 && fy <= nn && fy >= 1)
        {
    
    
            if(dp[fx][fy] == s[num])
            {
    
    
                if(i != flag)
                {
    
    
                    dfs(fx, fy, sum + 1, i, num + 1);
                }
                else
                {
    
    
                    dfs(fx, fy, sum, i, num + 1);
                }
            }
        }
    }
}
void solve()
{
    
    
    cin >> n >> nn;
    for(int i = 1; i <= n; i++)
    {
    
    
        for(int j = 1; j <= nn; j++)
        {
    
    
            cin >> dp[i][j];
        }
    }
    cin >> m;
    cin >> s;
    ans = 0;
    len = s.size();
    if(len == 1)
    {
    
    
        for(int i = 1; i <= n; i++)
        {
    
    
            for(int j = 1; j <= nn; j++)
            {
    
    
                if(dp[i][j]==s[0]&&m==0){
    
    
                    puts("YES");
                    return ;
                }
            }
        }
    }
    else
    {
    
    
        for(int i = 1; i <= n; i++)
        {
    
    
            for(int j = 1; j <= nn; j++)
            {
    
    
                if(dp[i][j] == s[0])
                {
    
    
                    dfs(i, j, -1, -1, 1);
                }
                if(ans == 1)
                {
    
    
                    puts("YES");
                    return ;
                }
            }
        }
    }
    puts("NO");
}
 
int main()
{
    
    
    solve();
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_45911397/article/details/115032768