POJ 3126 Prime Path 【bfs】

传送门
题意: 给定两个四位数的素数, 然后每次可以将第一个素数的某一位数字变成其他数字, 但是要保证变化后的那个数依旧是素数, 问最少需要多少步可以变成第二素数.

很明显的bfs, 先预处理下4位数的素数, 然后直接做bfs即可, 注意就是一些技巧为了书写方便.

AC Code

const int maxn = 1e4+5;
bool pri[maxn], vis[maxn];
void init() {
    for (int i = 1000 ; i < 10000 ; i ++) {
        if (i%2 == 0) {
            pri[i] = false;
            continue;
        }
        int f = 0;
        for (int j = 3 ; j * j <= i ; j +=2) {
            if (i % j == 0) {
                f = 1;
                break;
            }
        }
        if (f) pri[i] = false;
        else pri[i] = true;
    }
}
int stoi(string s) {
    int ans = 0, rat = 1;
    for (int i = 3 ; i >= 0 ; i --) {
        ans += (s[i] - '0' ) * rat;
        rat *= 10;
    }
    return ans;
}
string st, ed;
#define psi pair<string, int>
void solve()
{
    cin >> st >> ed; Fill(vis, 0);
    queue<psi>q; q.push({st, 0});
    vis[stoi(st)] = true;
    while(!q.empty()) {
        psi u = q.front();
        if (u.fi == ed) {
            printf("%d\n", u.se);
            break;
        }
        q.pop(); int mod = 1;
        string ss = u.fi;
        for (int i = 3 ; i >= 0 ; i --) {
            mod *= 10; u.fi = ss;
            if (i == 0) {
                for (char j = '1' ; j <= '9' ; j ++) {
                    if (j == u.fi[i]) continue;
                    u.fi[i] = j; int tmp = stoi(u.fi);
                    if (!pri[tmp] || vis[tmp]) continue;
                    vis[tmp] = 1;
                    q.push({u.fi, u.se+1});
                }
            }
            else {
                for (char j = '0' ; j <= '9' ; j ++) {
                    if (j == u.fi[i]) continue;
                    u.fi[i] = j; int tmp = stoi(u.fi);
                    if (!pri[tmp] || vis[tmp]) continue;
                    vis[tmp] = 1;
                    q.push({u.fi, u.se+1});
                }
            }
        }
    }
}