A*与IDA*

谨以此文向人工智能先驱,\(A\)*算法发明者\(Nils\ Nilsson\)致敬

一篇深入研究的博客,而本文更多是粗略理解和习题吧。

\(A\)*算法是什么?它是启发式搜索的一种,即广度搜索算法\(bfs\)加上估价函数。

\(IDA\)*则是另一种类似的启发式搜索,是迭代加深\(dfs\)加上估价函数。

迭代加深搜索是什么?就是每次限制\(dfs\)的深度进行搜索。

然后我们来讨论该算法的核心部分:估价函数

估价函数假设用\(h(x)\)表示,到达目前状态的代价用\(g(x)\)表示,那么估计的完成代价为\(f(x)=g(x)+h(x)\)。如果\(f(x)\)大于题目所需,那么可以进行剪枝。

需要注意的是,\(h(x)\)必须小于等于实际最优代价,满足此条件下越接近实际代价该估价函数越优。

一道例题:SCOI2005 骑士精神

对于一种状态,最佳的方案(运气最好时)是每次都能使一个骑士到达预定位置,最后一次能使一个骑士和一个空格到达预定位置,所以估价函数\(h(x)=\)棋盘上与目标状态不同的位置数\(+1\)

然后搜索,注意每次改变空格的位置即可。

提供一份\(IDA\)*的代码

#include<bits/stdc++.h>
using namespace std;
const int dx[]={-2,-2,-1,-1,1,1,2,2}, dy[]={-1,1,-2,2,-2,2,-1,1};
const int tar[5][5]=
{
    {2, 2, 2, 2, 2},
    {1, 2, 2, 2, 2},
    {1, 1, 0, 2, 2},
    {1, 1, 1, 1, 2},
    {1, 1, 1, 1, 1}
};
int sta[5][5];

inline int read()
{
    int x=0,f=1;char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar()) if (ch=='-') f=-1;
    for (;ch>='0'&&ch<='9';ch=getchar()) x=(x<<1)+(x<<3)+ch-'0';
    return x*f;
}

int evoluate()
{
    int res=0;
    for (int i=0; i<5; i++)
        for (int j=0; j<5; j++)
            if (tar[i][j]^sta[i][j]) res++;
    return res;
}

bool IDA(int x, int y, int dep, int maxdep)
{
    int evo=evoluate();
    if (dep+evo>maxdep+1) return false;
    if (!evo) return true;
    for (int i=0; i<8; i++)
    {
        int nx=x+dx[i], ny=y+dy[i];
        if (nx>=0 && nx<5 && ny>=0 && ny<5)
        {
            swap(sta[x][y], sta[nx][ny]);
            if (IDA(nx, ny, dep+1, maxdep)) return true;
            swap(sta[x][y], sta[nx][ny]);
        }
    }
    return false;
}

int main()
{
    int T=read();
    while (T--)
    {
        char s[5][5]; int X, Y;
        for (int i=0; i<5; i++) scanf("%s", s[i]);
        for (int i=0; i<5; i++)
            for (int j=0; j<5; j++)
            {
                char c=s[i][j]; 
                if (c=='0') sta[i][j]=1;
                if (c=='1') sta[i][j]=2;
                if (c=='*') X=i, Y=j, sta[i][j]=0;
            }
        int ans=-1;
        for (int i=1; i<=15; i++)
            if (IDA(X, Y, 0, i)) {ans=i; break;}
        printf("%d\n", ans);
    }
    return 0;
}

一道类似的题:八数码难题

本题可以用\(IDA\)*和双向\(bfs\)通过,甚至\(map\)去重的\(bfs\)也可通过。作为一道练手题还是不错的。

代码不放了,和上题大体一致(我才不会说是我懒)

再看一个\(A\)*经典应用:\(K​\)短路

(洛谷的模板题\(A​\)*只能得到\(92pts​\),标算是可持久化可并堆,所以这里就当是理性愉悦了吧)

说一下\(A​\)*算法的思路。

先在反向图中\(Dijkstra\)得到估价函数\(dis(i)\)

然后\(bfs\),用堆维护一下目前距离\(f(i)+dis(i)\)就可以了。

有一个没有什么用的优化,遍历每个点的次数不会超过\(\frac{w}{dis(1)}\),可以剪枝。

\(92pts\)代码

#include<bits/stdc++.h>
using namespace std;
const int N=5005, M=200005;
struct node{int to, nxt; double w;}edge1[M], edge2[M];
struct Node{int id; double w;};
struct NODE{int id; double w, f;};
bool operator < (Node a, Node b){return a.w>b.w;}
bool operator < (NODE a, NODE b){return a.f>b.f;}
int head1[N], head2[N], vis[N], cnt[N], cnt1, cnt2, n, m, ans;
double W, dis[N];

inline int read()
{
    int x=0,f=1;char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar()) if (ch=='-') f=-1;
    for (;ch>='0'&&ch<='9';ch=getchar()) x=(x<<1)+(x<<3)+ch-'0';
    return x*f;
}

void add(int u, int v, double w)
{
    edge1[++cnt1]=(node){v, head1[u], w};
    head1[u]=cnt1;
    edge2[++cnt2]=(node){u, head2[v], w};
    head2[v]=cnt2;
}

void Dijkstra(int S)
{
    priority_queue<Node> q; q.push((Node){S, 0});
    for (int i=1; i<=S; i++) dis[i]=1e12; dis[S]=0.0;
    memset(vis, 0, sizeof(vis));
    while (!q.empty())
    {
        int u=q.top().id; q.pop();
        if (vis[u]) continue; vis[u]=1;
        for (int i=head2[u]; i; i=edge2[i].nxt)
        {
            int v=edge2[i].to; double w=edge2[i].w;
            if (dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                q.push((Node){v, dis[v]});
            }
        }
    }
}

void A_star(int S, int Max_cnt)
{
    priority_queue<NODE> q; q.push((NODE){S, 0, 0});
    while (!q.empty())
    {
        NODE u=q.top(); q.pop(); int x=u.id;
        if (u.w>W) return;
        cnt[x]++;
        if (x==n) {ans++; W-=u.f; continue;}
        if (cnt[x]>Max_cnt) continue;
        for (int i=head1[x]; i; i=edge1[i].nxt)
        {
            int v=edge1[i].to;
            q.push((NODE){v, u.w+edge1[i].w, u.w+edge1[i].w+dis[v]});
        }
    }
}

int main()
{
    n=read(); m=read(); scanf("%lf", &W);
    for (int i=1; i<=m; i++)
    {
        int u=read(), v=read(); 
        double w; scanf("%lf", &w);
        add(u, v, w);
    }
    Dijkstra(n); A_star(1, W/dis[1]);
    printf("%d\n", ans);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/ACMSN/p/10771437.html