2024年牛客周赛round35 补题

D、小红的排列构造

思路：注意看清排列的定义。找出现次数大于一的， 把它转为没出现过的。

#include<bits/stdc++.h>

using namespace std;
const int N=100010;
int q[N],cnt[N];
int n;

int main(){
    cin>>n;
    for(int i=1;i<=n;i++) 
    {
        cin>>q[i];
        if(q[i]<=n) cnt[q[i]]=1;
    }
    vector<pair<int,int>>v;//用二元组存修改答案
    
    for(int i=1,j=1;i<=n;i++){//遍历数组，如果出现次数为1，就修改次数，使之在下一次出现时能够进入修改。
        if(q[i]<=n&&cnt[q[i]]==1){
            cnt[q[i]]=2;
            continue;
        }
        while(cnt[j]!=0) j++;//寻找1~n空的地方才能放
        v.push_back({i,j});
        cnt[j]=1;//把空的地方，状态修改为不空
    }
    cout<<v.size()<<endl;
    
    for(auto i:v) cout<<i.first<<' '<<i.second<<'\n';
    return 0;
}

E、 小红的无向图构造

思路：可以先根据最短路构造树，再慢慢改为图。

树：n个点有n-1条边；但现在要m条边，所以要加边

第一种情况：加边可以在距离点1相同距离的点之间加（dis[i]==dis[j]）

第二种情况：距离相等的点中，某个点后有一个点，该点也可以连其他与某个点相同距离的点(abs(dis[i]-dis[j])==1)

否则会破坏最短路。

特判：每一个点距离点1的距离大小应该是连续的

#include<bits/stdc++.h>

using namespace std;
int a[202002];
vector<int>g[202020];//用于把相同值的数存到相同的数组

int main(){
    int n,m;
    cin>>n>>m;
    int ma=0;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        ma=max(ma,a[i]);
        g[a[i]].push_back(i);
    }
    if(m<n-1){//边的数量无法生成树
        cout<< -1;
        return 0;
    }
    //连边
    vector<pair<int,int>>v; 
    for(int i=1;i<=ma;i++){
        if(g[i].empty()){
            cout<< -1;
            return 0;
        }
        for(auto j:g[i]){
            v.push_back({g[i-1][0],j});
        }
    }
    if(v.size()<m){
        for(int i=1;i<=ma;i++){
            for(int j=0;j<g[i].size();j++){
                for(int k=j+1;k<g[i].size();k++){
                     v.push_back({g[i][j],g[i][k]});
                    if(v.size()==m)break;
                }
                if(v.size()==m)break;
            }
            if(v.size()==m)break;
        }
        for(int i=1;i<=ma;i++){
            for(int j=1;j<g[i].size();j++){
                for(auto k:g[i+1]){
                    v.push_back({g[i][j],k});
                    if(v.size()==m)break;
                }if(v.size()==m)break;
            }
            if(v.size()==m)break;
        }
        
    }
    if(v.size()<m){
        cout<< -1;
        return 0;
    }
    for(auto i:v)cout<<i.first<<" "<<i.second<<'\n';
    
}