POJ 3126 Prime Path（素数，BFS最短路）

【链接】http://poj.org/problem?id=3126

【题意】给个字符串，求满足形如“E...E...E”这种鬼样子的E串最大长度，其中“...”可以是任意个数的任意字符(没有也ok)，但是三个E之间不能重叠

【思路】

 ① 掏出素数筛的模板，找到1W以内素数

 ② 留下1000~9999之间的保存在prime[]，用个map记录一下下标

 ③ 因为一起也就2000多点素数，n^2处理一下之间是否可以一步到达，用vector<int>的数组存下邻接表

 ④ 读素数，mp出下标，最基础的BFS来一发找出最短路（调1个数字就能转换的数之间距离为1）就是了

【代码】作大死，套了半天dijkstra调不好还妄图使用spfa最后还是老老实实bfs一下……

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <iostream>
#include <map>
#include <vector>
#include <set> 
#include <queue>
#include <stack>
#include <algorithm>
using namespace std;

#define MAX_LEN 2100
#define ll long long 
#define mod 100000007

#define MEM(a,al) memset(a,al,sizeof(a))
#define sfx(x) scanf("%lf",&x)
#define sfxy(x,y) scanf("%lf%lf",&x,&y)
#define sdx(x) scanf("%d",&x)
#define sdxy(x,y) scanf("%d%d",&x,&y)
#define pfx(x) printf("%.0f\n",x)
#define pfxy(x,y) printf("%.6f %.6f\n",x,y)
#define pdx(x) printf("%d\n",x)
#define pdxy(x,y) printf("%d %d\n",x,y)
#define getArray(a,len) for(int ia = 0; ia < len; ia++) scanf("%d",&a[ia])
#define printArray(a,len) for(int ia = 0; ia < len; ia++) printf("%d%c",a[ia],(ia==len-1)?'\n':' ')
#define fora(i,n) for(i = 0; i < n; i++)
#define fora1(i,n) for(i = 1; i <= n; i++)
#define foraf(i,n) for(int i = 0; i < n; i++)
#define foraf1(i,n) for(int i = 1; i <= n; i++)
#define ford(i,n) for(i = n-1; i >= 0; i--)
#define ford1(i,n) for(i = n; i > 0; i--)
#define fordf(i,n) for(int i = n-1; i >= 0; i--)
#define fordf1(i,n) for(int i = n; i > 0; i--)

class WriteInfo{
public:
    void info() { printf("[Info] "); }
    void infoEnd() { printf(" [/Info]\n"); }
} W;

const int INF = 1<<29;
const double INFD = 1e20;
const double eps =  1e-6;


int n,m,a,b;
map< int, int > mp;
vector< int > v[MAX_LEN];
int prime[MAX_LEN];

// 素数筛选，素数存放于st （此模板来源于网络……）
set<int> st;  
#define MAX_P  10000  
int nList[MAX_P] = {0};  
void Calc()   {  
    int n,p,t,sq=(int)sqrt(MAX_P*2+1);  
    for (n=3;n<=sq;n+=2) {  
        if (nList[n>>1]) continue;  
        for (t=n*n;t<=MAX_P<<1;t+=n<<1) //筛选循环  
            nList[t>>1] = 1;  
    }  
    st.insert(2);  
    for (n=t=1;t<MAX_P;++t) {  
        if (nList[t]) continue;  
        st.insert((t<<1)+1);  
        if (++n==10)  
            n=0;  
    }  
}  

bool check(int x, int y){
    int res = 0;
    while(x){
        if (x%10 == y%10)
            res++;
        x /= 10;
        y /= 10;
    }
    return (res == 3);
}

int bfs(int x, int y){
    if (x == y) return 0;
    int dis[MAX_LEN];
    MEM(dis, -1);
    queue< int > q;
    q.push(x);
    dis[x] = 0;
    while(!q.empty()){
        int p = q.front();
        q.pop();
        foraf(i,v[p].size()){
            int next = v[p][i];
            if (next == y) 
                return dis[p] + 1;  
            if (dis[next] == -1){
                q.push(next);
                dis[next] = dis[p] + 1;
            }
        }
    }
    return -1;
}

int main() {
    int i,j,k;
    int T; 

    // 掏来的模板筛素数，放到mp里映射一下保存prime数组中的下标
    Calc();
    mp.clear();
    int co = 0;
    for(set< int >::iterator it = st.begin(); it != st.end(); it++){
        if (*it > 1000 && *it < 10000){
            mp[*it] = co;
            prime[co++] = *it;
        }
    }

    // 预处理，总共也就2000+个素数
    fora(i,co)
        for(j = i+1; j < co; j++)
            if (check(prime[i],prime[j])){
                v[i].push_back(j);
                v[j].push_back(i);
            }


    sdx(T);
    while(T--){
        sdxy(a,b);
        // 直接用下标来做
        int res = bfs(mp[a], mp[b]);
        if (res != -1)
            pdx(res);
        else
            puts("Impossible");
    }  

    return 0;
}

为写一篇日志做了整整一个月半的心理准备【滚！