2020 NUC 19级第一次训练赛

感染(low)

Description

 

n户人家住在一条直线上,从左往右依次编号为1,2,...,n。起初,有m户人家感染了COVID-19,而接下来的每天感染的人家都会感染他家左右两家的人,问t天后总共有多少人家会感染。

 

Input

 

第一行输入三个整数n(1 <= n <= 2e5),m(0 <= m <= n),t(1<= t <= 2e5)。

第二行m个整数ai(1 <= ai <= n),代表最初感染的m户人家的编号。

 

Output

 

输出一个整数,代表最后会有多少人家感染。

 

Sample Input 1 

10 3 2
1 3 4

Sample Output 1

6

分析:这个题实际上就是以m个点为中心,生成了m条长度最多为2*m的线段,然后合并这m条线段就可以了。
第i条线段的信息可以表示为[max(1, a[i] - t), min(n, a[i] + t)}。

代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 7;
int a[maxn];
vector<pair<int, int> > v;
int main() {
    int n, m, t;
    cin >> n >> m >> t;
    for (int i = 1; i <= m; i++)
        cin >> a[i];
    sort(a + 1, a + 1 + m);
    for (int i = 1; i <= m; i++) {
        int l = max(1, a[i] - t);
        int r = min(n, a[i] + t);
        v.push_back(make_pair(l, r));
    }
    int i = 0, res = 0;
    while (i < m) {
        int j = i + 1;
        while (j < m && v[j].first <= v[j - 1].second) {
            j++;
        }
        res += v[j - 1].second - v[i].first + 1;
        i = j;
    }
    cout << res <<endl;
    return 0;
}

第k短路
 

Description

 

一天,HighLights实在是闲的不行,他选取了n个地点,n各地点之间共有m条路径,他想找到这m条路径组成的第k短路,你能帮助他嘛?

 

Input

 

第一行三个正整数,地点的数量n(2 <= n <= 2e5),边的数量m(1 <= m <= 2e5),k(1 <= k <= min(m,  200))。

接下来m行,每行三个整数,边的一个顶点u(1<=u<=n),边的另一个顶点v(1<=v<=n),边的权值w(1<=w<=1e5),代表u有一条到v权值为w的单向边。

 

Output

 

输出第k短路的权值。

 

Sample Input 1 

4 4 3
1 3 27
1 4 16
1 2 15
2 4 3

Sample Output 1

16

分析:这个题可以刚开始看的时候被吓到了,但是实际上这个题比较简单。因为k被限制了,k比较小,保证了k不大于m。
也就是第k短路是由权值前k小的边产生的子图构成的。

那么我们就可以对边进行权值从小到大排序,然后取权值前k的边,重新建图。
这里对顶点需要进行重新编号,可以使用map或set,然后建立一个新的单向图。
再对这个图跑个floyd,最后把G[i][j](i != j)都取出来,

从小到大排个序,那么排序完后第k个就是我们想要的第k短路了。
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn = 4e5 + 7;
const int maxm = 4e2 + 7;
struct node{
    int u, v, w;
} e[maxn];
bool cmp(node a, node b) {
    return a.w < b.w;
}
set<int> st;
int G[maxm][maxm];
map<int, int> id;
vector<int> res;
int main() {

    int n, m, t;
    memset(G, 0x3f, sizeof(G));
    for (int i = 0; i < maxm; i++) {
        G[i][i] = 0;
    }
    cin >> n >> m >> t;
    for (int i = 0; i < m; i++) {
        cin >> e[i].u >> e[i].v >> e[i].w;
    }
    sort(e, e + m, cmp);
    for (int i = 0; i < t; i++) {
        st.insert(e[i].u);
        st.insert(e[i].v);
    }
    int len = 0;
    for (auto p = st.begin(); p != st.end(); p++) {
        id[*p] = ++len;
    }
    for (int i = 0; i < t; i++) {
        G[id[e[i].u]][id[e[i].v]] = min(G[id[e[i].u]][id[e[i].v]], e[i].w);
    }
    for (int i = 1; i <= len; i++) {
        for (int j = 1; j <= len; j++) {
            for (int k = 1; k <= len; k++) {
                G[i][j] = min(G[i][j], G[i][k] + G[k][j]);
            }
        }
    }
    for (int i = 1; i <= len; i++) {
        for (int j = 1; j <= len; j++) {
            if (i == j) continue;
            res.push_back(G[i][j]);
        }
    }
    sort(res.begin(), res.end());
    cout << res[t - 1] << endl;
    return 0;
}

 

posted @ 2020-02-23 11:56  SwiftAC  阅读(312)  评论(0编辑  收藏  举报