素数判断+BFS POJ 3126

题意：

给两个四位的素数，求出从第一个素数变为第二个素数的最短路径。每步可以变素数的一位，并且每步得到的数必须是素数。

先把素数打表，然后bfs求最短路径就可以了，如果变换后得到的数是素数，就加入队列。

代码：

#include <cstdlib>
#include <cctype>
#include <cstring>
#include <cstdio>
#include <cmath>
#include<climits>
#include <algorithm>
#include <vector>
#include <string>
#include <iostream>
#include <sstream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <fstream>
#include <numeric>
#include <iomanip>
#include <bitset>
#include <list>
#include <stdexcept>
#include <functional>
#include <utility>
#include <ctime>
using namespace std;

#define PB push_back
#define MP make_pair

#define REP(i,x,n) for(int i=x;i<(n);++i)
#define FOR(i,l,h) for(int i=(l);i<=(h);++i)
#define FORD(i,h,l) for(int i=(h);i>=(l);--i)
#define SZ(X) ((int)(X).size())
#define ALL(X) (X).begin(), (X).end()
#define RI(X) scanf("%d", &(X))
#define RII(X, Y) scanf("%d%d", &(X), &(Y))
#define RIII(X, Y, Z) scanf("%d%d%d", &(X), &(Y), &(Z))
#define DRI(X) int (X); scanf("%d", &X)
#define DRII(X, Y) int X, Y; scanf("%d%d", &X, &Y)
#define DRIII(X, Y, Z) int X, Y, Z; scanf("%d%d%d", &X, &Y, &Z)
#define OI(X) printf("%d",X);
#define RS(X) scanf("%s", (X))
#define MS0(X) memset((X), 0, sizeof((X)))
#define MS1(X) memset((X), -1, sizeof((X)))
#define LEN(X) strlen(X)
#define F first
#define S second
#define Swap(a, b) (a ^= b, b ^= a, a ^= b)
#define Dpoint  strcut node{int x,y}
#define cmpd int cmp(const int &a,const int &b){return a>b;}

 /*#ifdef HOME
    freopen("in.txt","r",stdin);
    #endif*/
const int MOD = 1e9+7;
typedef vector<int> VI;
typedef vector<string> VS;
typedef vector<double> VD;
typedef long long LL;
typedef pair<int,int> PII;
//#define HOME

int Scan()
{
	int res = 0, ch, flag = 0;

	if((ch = getchar()) == '-')				//判断正负
		flag = 1;

	else if(ch >= '0' && ch <= '9')			//得到完整的数
		res = ch - '0';
	while((ch = getchar()) >= '0' && ch <= '9' )
		res = res * 10 + ch - '0';

	return flag ? -res : res;
}
/*----------------PLEASE-----DO-----NOT-----HACK-----ME--------------------*/

int prime[10000];
int vis[10005];
void getprime()
{int cnt=0;
    for(int i=2;i<=10000;i++)
        {
            if(!vis[i])
            {
                prime[cnt++]=i;

            }
            for(int j=0;j<cnt&&prime[j]<=10000/i;j++)
            {
                vis[prime[j]*i]=1;
                if(i%prime[j]==0)
                    break;
            }
        }
}

string str1,str2;
queue<string>q;
set<string>r;
int cost[100000];
int bfs()
{while(!q.empty())
q.pop();
int front=0,rear=1;
q.push(str1);
r.clear();
r.insert(str1);
cost[0]=0;
while(front<rear)
{
    string now=q.front();
    q.pop();
    if(now==str2)
        return cost[front];
    REP(i,0,4)
    REP(j,0,10)
    {if(i==0&&j==0)
    continue;
    string tmp=now;
    tmp[i]='0'+j;
    int res=0;
    REP(k,0,4)
    res=res*10+tmp[k]-'0';
    if(!vis[res]&&!r.count(tmp))
    {q.push(tmp);
    r.insert(tmp);
    cost[rear]=cost[front]+1;
    rear++;

    }

    }
    front++;
}
return -1;

}
int main()
{getprime();
int T;
RI(T);
while(T--)
{cin>>str1>>str2;
MS0(cost);
int ans=bfs();
if(ans!=-1)
    printf("%d\n",ans);
else
    printf("Impossible\n");

}

        return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。