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

题目链接

可能就不去~~~

题意:

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

思路:

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

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

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
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;
}

你可能感兴趣的:(bfs)