2024年2月19日 牛客周赛round33 补题

C、小红的01串

思路：贪心、后缀和。可以把最终答案分为两部分，根据题目可以发现最终答案第一位始终为1，所以答案可以分为第一位和后面部分，只需要取后缀的最大值再加1就是答案。

#include<bits/stdc++.h>
using namespace std;
int main(){
    string s;
    cin>>s;
    int n=s.length(),i,j;
    for(i=0;i<n;i++){//寻找第一个1
        if(s[i]=='1')break;
    }
    if(i==n) {//说明全部为0
        cout<<0;
        return 0;
    }
    int res=1,temp=0;//res负责记录最大值，temp负责记录当前值
    for(j=n-1;j>i;j--){
        if(s[j]=='0')temp--;
        else temp++;
        res=max(res,temp+1);
    }
    cout<<res;
    return 0;
}

D、 小红的数组清空

思路：可以先排序，从最小的开始删除，只要存在x+1就继续往后删。注意：这里需要使用map之类的容器计算出现次数。

#include<bits/stdc++.h>
#include<algorithm>
using namespace std;
const int N=100010;


int main(){
    int n,d=0;
    cin>>n;
    vector<int>q(n);
    map<int,int> count;
    for(int i=0;i<n;i++) {
        cin>>q[i];
        count[q[i]]++;
    }
    sort(q.begin(),q.end());//(里面填地址)
    
    for(int i=0;i<n;i++){
        int tmp=q[i];
        if(count[q[i]]==0) continue;
        count[q[i]]--;
        while(count[tmp+1]>0){
            count[tmp+1]--;
            tmp++;
        }
        d++;
    }
    printf("%d",d);
    return 0;
}

E、小红勇闯地下城

思路：Dijkstra算法。

#include<bits/stdc++.h>
using namespace std;
string s[101010];
vector<pair<int,int>>g[101010];
int op[4][2]={0,1,0,-1,1,0,-1,0};
long long dis[101010];
struct node{
    int x;
    node(int x):x(x){}
    bool operator<(const node &x)const{
        return dis[this->x]>dis[x.x];
    }
};
int main(){
    int a;
    cin>>a;
    while(a--){
        int n,m,h,i,j,k;
        cin>>n>>m>>h;
        for(i=0;i<n;i++)cin>>s[i];
        for(i=0;i<n;i++){
            for(j=0;j<m;j++){
                g[i*m+j].clear();
                dis[i*m+j]=1e15;
            }
        }
        int x,y,xx,yy;
        for(i=0;i<n;i++){
            for(j=0;j<m;j++){
                if(s[i][j]=='S')x=i,y=j,s[i][j]='0';
                else if(s[i][j]=='T')xx=i,yy=j,s[i][j]='0';
                for(k=0;k<4;k++){
                    int tx=i+op[k][0],ty=j+op[k][1];
                    if(tx<0||tx==n||ty<0||ty==m)continue;
                    g[tx*m+ty].push_back({i*m+j,s[i][j]-'0'});
                }
            }
        }
    
        priority_queue<node>q;
        dis[x*m+y]=0;
        q.push(node(x*m+y));
        
        while(!q.empty()){
            node temp=q.top();
            q.pop();
            for(auto i:g[temp.x]){
                int u=i.first,v=i.second;
                if(dis[u]>dis[temp.x]+v){
                    dis[u]=dis[temp.x]+v;
                    q.push(u);
                }
            }
        }
        
        if(dis[xx*m+yy]>=h)cout<<"No\n";
        else cout<<"Yes\n";
        
        
        //i行j列节点的编号是 i*m+j
    }
}