程序设计实习上机练习43 至少有多少只恼人的大青蛙?

枚举路线即可，加上dfs过程中的剪枝，优化的速度很快

#include<iostream>
#include<queue>
#include<cstring>
#include<string>
#include<algorithm>
#include<set>
using namespace std;
int r,c,rnum;
int sum;
int n;
int ma[53][53]={};
int ans=14;
struct point{
int x;
int y;
bool operator<(const point &a)const{
if(a.x==x)
return y<a.y;
return x<a.x;
}
point(){};
}p[702];
struct route{
point start;
int dx;
int dy;
int step;
bool operator<(const route &a)const{
return step>a.step;
};
route(point a,int b,int c,int d){
start=a;
dx=b;
dy=c;
step=d;
};
route(){};
}ro[1000001];
bool inside(int x,int y){
return (1<=x && x<=r && 1<=y && y<=c);
}
int match(point start,int dx,int dy){
       if (inside(start.x-dx,start.y-dy))
               return 0;
       int k;
            for (k=0;;k++)
            {
                if (inside(start.x+k*dx,start.y+k*dy))
                {
                    if (ma[start.x+k*dx][start.y+k*dy]==0)
                    {
                        return 0;
                    }
                }
                else
                    break;
            }
if(k>2)
return k;
return 0;
}
void dfs(int code,int s,int num){
if(sum==s){
ans=min(ans,num);
return;
}
for(int i=code;i<rnum;++i){
if(num+(sum-s)/ro[i].step>=ans)return;
if(match(ro[i].start,ro[i].dx,ro[i].dy)==0)continue;
for(int x=ro[i].start.x,y=ro[i].start.y;1<=x&&x<=r&&y>=1&&y<=c;x+=ro[i].dx,y+=ro[i].dy)--ma[x][y];
dfs(i,s+ro[i].step,num+1);
for(int x=ro[i].start.x,y=ro[i].start.y;1<=x&&x<=r&&y>=1&&y<=c;x+=ro[i].dx,y+=ro[i].dy)++ma[x][y];
}
return;
}
int main(){
int t;
cin>>t;
while(t--){
rnum=0;
ans=14;
cin>>r>>c;
cin>>n;
sum=0;
for(int i=0;i<n;++i){
cin>>p[i].x>>p[i].y;
int l;
cin>>l;
sum+=l;
ma[p[i].x][p[i].y]=l;
}
sort(p,p+n);
for(int i=0;i<n;++i){
for(int j=i+1;j<n;++j){
int st=match(p[i],p[j].x-p[i].x,p[j].y-p[i].y);
if(st!=0){
ro[rnum++]=route(p[i],p[j].x-p[i].x,p[j].y-p[i].y,st);
}
}
}
sort(ro,ro+rnum);
dfs(0,0,0);
cout<<ans<<endl;
}
}