多源bfs

最新推荐文章于 2024-06-02 13:10:50 发布

zipper112

最新推荐文章于 2024-06-02 13:10:50 发布

阅读量475

点赞数

分类专栏： Codeforces 搜索

本文链接：https://blog.csdn.net/qq_36102055/article/details/105326390

版权

Codeforces 同时被 2 个专栏收录

5 篇文章 0 订阅

订阅专栏

搜索

3 篇文章 0 订阅

订阅专栏

题目一

题意：

定义一种数为Lunlun数，这种数的任意相邻的位上的数 $a_i,a_{i+1}$ 都有 $abs(a_i - a_{i+1})\le1$ ，例：121001，现在给你一个正整数k让你求，从1开始的第k个Lunlun数

思路：

k的范围很小，所以可以用多源bfs，首先要先确定范围即第1e5个数最大可能是多少，或者知道我们需要bfs多少次可以得到第1e5个数，第一种方法可以直接推(我不会)，所以我用第二种，先来看bfs：

for(int i = 0; i < 9; i++)
        qu.push(i + 1);
    while(qu.size()){
        ll t = qu.front();
        qu.pop();
        if(v.size() > k * 9)
            continue;
        v.push_back(t);
        ll temp2 = t % 10;
        if(temp2 == 9){
            qu.push(t * 10 + 9);
            qu.push(t * 10 + 8);
        }
        else if(temp2 == 0){
            qu.push(t * 10);
            qu.push(t * 10 + 1);
        }
        else{
            qu.push(t * 10 + temp2);
            qu.push(t * 10 + temp2 - 1);
            qu.push(t * 10 + temp2 + 1);
        }
    }

可以发现我取的bfs数是 $k * 9$ ，多源bfs进行k次不一定能求出前k个数，因为他们的顺序不一，但是我们初始进行bfs的根是1~9个数，如果他们都bfs进行k次则一定包含前k个数。

#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
#define eps 1e-8
using namespace std;
const ll maxn = 1e6 + 5;
ll a,b,c,d,k,n,m,t,num[maxn];
ll ans[maxn];
queue<ll> qu;
vector<ll> v;
set<ll> st;
void slove(){
    cin >> k;
    for(int i = 0; i < 9; i++)
        qu.push(i + 1);
    while(qu.size()){
        ll t = qu.front();
        qu.pop();
        if(v.size() > k * 9)
            continue;
        v.push_back(t);
        ll temp2 = t % 10;
        if(temp2 == 9){
            qu.push(t * 10 + 9);
            qu.push(t * 10 + 8);
        }
        else if(temp2 == 0){
            qu.push(t * 10);
            qu.push(t * 10 + 1);
        }
        else{
            qu.push(t * 10 + temp2);
            qu.push(t * 10 + temp2 - 1);
            qu.push(t * 10 + temp2 + 1);
        }
    }
}
int main()
{
    slove();
    sort(v.begin(),v.end());
    cout << v[k - 1] << endl;
    return 0;
}

题目二

题意：

给定n个互不相等整数 $x_1,x_2...x_n$ ，求另m个整数 $y_1,y_2...y_m$ ，满足：
1.任意 $x\neq y$ 且任意y不相等
2.定义 $d_i = min|y_1 - x_j|$ ，使 $\sum\limits_{i = 1}^{m}di$ 最小

思路：

同样的思路我们可以先贪心，对于 $x_i$ 我们可以让 $y = x_i+1,y = x_i - 1$ 再来 $x_i - 2,x_i+ 2...$ 这样 $d_i$ 就从1开始一点一点大也就让 $\sum\limits_{i = 1}^{m}di$ 最小了

#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
#define eps 1e-8
using namespace std;
const ll maxn = 1e6 + 5;
ll a,b,c,d,k,n,m,t,num[maxn],sum;
typedef pair<int,int> PL;
set<int> st;
vector<int> v;
queue<PL> qu;
int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
        cin >> num[i];
    for(int i = 1; i <= n; i++){
        st.insert(num[i]);
        qu.push({num[i],num[i] - 1}),qu.push({num[i],num[i] + 1});
    }
    while(qu.size() && m > 0){
        PL t = qu.front();
        qu.pop();
        if(st.find(t.second) != st.end())
            continue;
        st.insert(t.second),sum += abs(t.first - t.second);
        m--,v.push_back(t.second);
        qu.push({t.first,t.second - 1});
        qu.push({t.first,t.second + 1});
    }
    cout << sum << endl;
    for(int i = 0; i < v.size(); i++)
        cout << v[i] << " ";
    return 0;
}