NEUQacm第六周周报

P1135 奇怪的电梯

思路：模拟，广度优先搜索

#include <bits/stdc++.h>
using namespace std;
struct Node {
    int pos, sum;
};
queue <Node> q;
int n, a, b, d[205];
bool v[205];
int main() {
    cin >> n >> a >> b;
    for (int i = 1; i <= n; i++) {
        cin >> d[i];
    }
    Node h, t;
    h.pos = a;
    h.sum = 0;
    v[a] = true;
    q.push(h);
    while (!q.empty()) {
        h = q.front();
        if (h.pos == b) {
            cout << h.sum << endl;
            return 0;
        }
        for (int i = 1; i <= 2; i++) {
            if (i == 1)
                t.pos = h.pos + d[h.pos];
            else 
                t.pos = h.pos - d[h.pos];
            t.sum = h.sum + 1;
            if (t.pos < 1 || t.pos > n) continue;
            if (!v[t.pos]) {
                v[t.pos] = true;
                q.push(t);
            }
        }
        q.pop();
    }
    cout << -1 << endl;
    return 0;
}
P1443 马的遍历

思路：搜索，广度优先搜索

#include<bits/stdc++.h>
#include<queue>
using namespace std;
int n,m,x,y,t;//t为马的步数
int d[8][2]={ {-1,-2},{1,-2},{-2,-1},{2,-1},{-2,1},{2,1},{-1,2},{1,2}};
int l[500][500];
struct nodes{
    int x,y;
}node,top;
queue<nodes> q;
 
void bfs(){
    if(q.empty()) return ;
    top=q.front();
    q.pop();
    t=l[top.x][top.y]+1;
    for(int i=0;i<8;i++){
        int dx=top.x+d[i][0];
        int dy=top.y+d[i][1];
        if(dx<1||dy<1||dx>n||dy>m||l[dx][dy]) continue;
        node.x=dx;
        node.y=dy;
        q.push(node);
        l[dx][dy]=t;
    }
    bfs();
}
int main()
{
    cin>>n>>m>>x>>y;
    node.x=x;
    node.y=y;
    q.push(node);
    bfs();
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(i==x&&j==y)
               printf("%-5d",0);
            else if(l[i][j]==0)
               printf("%-5d",-1);
            else
               printf("%-5d",l[i][j]);
        }
        printf("\n");
    }
    return 0;
}

P3958 [NOIP2017 提高组] 奶酪

思路：搜索，并查集，广度优先搜索

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue>
#define ll long long
using namespace std;
const int N=1e3+5;
const int T=20;
 
struct POI{
    int x,y,z;
}p[N];
int vis[N];
int t;
int n,h,r;
 
ll dist(POI a,POI b){
    return (ll)(a.x-b.x)*(a.x-b.x)+(ll)(a.y-b.y)*(a.y-b.y)+(ll)(a.z-b.z)*(a.z-b.z);
}
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%d",&n,&h,&r);
        for(int i=1;i<=n;i++){
            scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].z);        
        }
        for(int i=1;i<=n;i++){
            vis[i]=0;
        }
        
        queue<int>q;
        for(int i=1;i<=n;i++){
            if(p[i].z-r<=0){
                vis[i]=1;
                q.push(i);
            }
        }
        while(!q.empty()){
            int u=q.front();
            q.pop();
            for(int i=1;i<=n;i++){
                if(i==u || vis[i]==1)continue;
                if(dist(p[u],p[i])<=(ll)4*r*r){
                    vis[i]=1;
                    q.push(i);
                }
            }
        }
        int f=0;
        for(int i=1;i<=n;i++){
            if(p[i].z+r>=h && vis[i]==1){
                f=1;
                break;
            }
        }
        if(f==1)puts("Yes");
        else puts("No");
    }
    return 0;
}

P1162 填涂颜色

思路：搜索，广度优先搜索

#include<bits/stdc++.h>
using namespace std;

int n;
int dx[]={1,0,0,-1};
int dy[]={0,1,-1,0};
int pos[35][35];

void dfs(int x,int y){
    if(pos[x][y]==1) return;//如果点是1，表示是墙，就返回，搜不进去了
    if(pos[x][y]==3) return;//如果=3，表示这个点已经搜过，也返回
    pos[x][y]=3;
    for(int k=0;k<4;k++){
        int tx=x+dx[k],ty=y+dy[k];
        if(tx>=0&&tx<=n+1&&ty>=0&&ty<=n+1){//注意：因为在外面加了一层，所以是0和n+1
            dfs(tx,ty);
        }
    }
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            cin>>pos[i][j];
    }
    dfs(0,0);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
        //如果点是3，就输出0，是1就输出1，是0就输出2.
            if(pos[i][j]==0) cout<<2<<" ";//如果是0，就表示当时搜索时遇到了1，然后返回，所以这里的0是围墙里面的0，输出2
            else if(pos[i][j]==1) cout<<1<<" ";
            else cout<<0<<" ";
        }
        cout<<endl;
    }
    return 0;
}