pku 2331 Water pipe 搜索+上下界剪枝

题意是这样的:
有一堆管子,长度为k,数量为n,然后起点为x1,y1,终点为x2,y2,管道只能水平、垂直放置,问最少需要的管子数量。如果不可能,输出-1
这题由于状态太大,采取一般的BFS肯定要爆时间,所以,还是用IDA*。首先,观察到x方向和y方向可以独立来处理,下限函数可以取为没有数量限制时从a到b需要的最少管道数,这个可以通过BFS来解决,复杂度为O(n)。
不多说,贴代码
  1Source Code 
  2# include <cstdio>
  3# include <cstring>
  4# include <queue>
  5using namespace std;
  6int c[5],l[5],n,res=0xfffffff;
  7int x1,y1,x2,y2;
  8int referx[1001],refery[1001];
  9queue<int> q;
 10bool solve(int pos,int len,bool type)
 11{
 12   if(!type)
 13   {
 14        if(referx[pos]==-1||len+referx[pos]>res) return false;
 15        if(pos==x2) 
 16        {
 17            return solve(y1,len,1);
 18        }

 19        for(int i=1;i<=n;i++)
 20          if(c[i])
 21          
 22            if(x2<pos)
 23            {
 24             if(pos-l[i]>=1)
 25             {
 26                c[i]--;
 27                if(solve(pos-l[i],len+1,0)) return true;;
 28                c[i]++;
 29             }

 30             if(pos+l[i]<=1000)
 31             {
 32                c[i]--;
 33                if(solve(pos+l[i],len+1,0)) return true;
 34                c[i]++;
 35             }

 36            }

 37            else
 38            {
 39             if(pos+l[i]<=1000)
 40             {
 41                c[i]--;
 42                if(solve(pos+l[i],len+1,0)) return true;
 43                c[i]++;
 44             }

 45             if(pos-l[i]>=1)
 46             {
 47                c[i]--;
 48                if(solve(pos-l[i],len+1,0)) return true;;
 49                c[i]++;
 50             }

 51            }

 52          }

 53   }

 54   else
 55   {
 56        if(refery[pos]==-1||len+refery[pos]>res) return false;
 57        if(pos==y2) 
 58        {
 59            return true;
 60        }

 61        for(int i=1;i<=n;i++)
 62          if(c[i])
 63          
 64            if(y2<pos)
 65            {
 66             if(pos-l[i]>=1)
 67             {
 68                c[i]--;
 69                if(solve(pos-l[i],len+1,1)) return true;
 70                c[i]++;
 71             }

 72             if(pos+l[i]<=1000)
 73             {
 74                c[i]--;
 75                if(solve(pos+l[i],len+1,1)) return true;
 76                c[i]++;
 77             }

 78            }

 79            else
 80            {
 81             if(pos+l[i]<=1000)
 82             {
 83                c[i]--;
 84                if(solve(pos+l[i],len+1,1)) return true;
 85                c[i]++;
 86             }

 87             if(pos-l[i]>=1)
 88             {
 89                c[i]--;
 90                if(solve(pos-l[i],len+1,1)) return true;
 91                c[i]++;
 92             }

 93            }

 94          }

 95   }

 96   return false;
 97}

 98void cal(int refer[],int pos)
 99{
100    refer[pos]=0;
101    q.push(pos);
102    while(!q.empty())
103    {
104       pos=q.front();
105       q.pop();
106       for(int i=1;i<=n;i++)
107       {
108         if(pos-l[i]>=1&&refer[pos-l[i]]==-1)
109          {
110              refer[pos-l[i]]=refer[pos]+1;
111              q.push(pos-l[i]);
112          }

113         if(pos+l[i]<=1000&&refer[pos+l[i]]==-1)
114          {
115              refer[pos+l[i]]=refer[pos]+1;
116              q.push(pos+l[i]);
117          }

118       }

119    }

120}

121int main()
122{
123    int total=0;
124    scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&n);
125    for(int i=1;i<=n;i++)
126       scanf("%d",l+i);
127    for(int i=1;i<=n;i++)
128    {
129       scanf("%d",c+i);
130       total+=c[i];
131    }

132    memset(referx,-1,sizeof(referx));
133    memset(refery,-1,sizeof(refery));
134    cal(referx,x2);
135    cal(refery,y2);
136    for(res=0;res<=total;res++)
137       if(solve(x1,0,0)) break;
138    if(res==total+1) printf("-1\n");
139    else printf("%d\n",res);
140 //   system("pause");
141    return 0;
142}

143
144

posted on 2010-10-19 14:44 yzhw 阅读(186) 评论(0)  编辑 收藏 引用 所属分类: search


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


<2010年10月>
262728293012
3456789
10111213141516
17181920212223
24252627282930
31123456

导航

统计

公告

统计系统

留言簿(1)

随笔分类(227)

文章分类(2)

OJ

最新随笔

搜索

积分与排名

最新评论

阅读排行榜