AtCoder Beginner Contest 304（D - A Piece of Cake & E - Good Graph）

D - A Piece of Cake

原题链接

D - A Piece of Cake

题面：

思路

被划分的块的数量与范围有$10^{10}$ 那么大，所以直接遍历块的数量是不现实的，但是草莓的数量的有限的，只有$2\ast 10^5$多，对每一个草莓，二分统计它在哪一块蛋糕上，用map计数即可。遍历map即可得到最大值。但是对于每一块蛋糕上的最小草莓数，如果蛋糕的数量大于草莓的数量，那么必定有一块蛋糕上草莓数为0，然后，其实可以二重循环遍历蛋糕数，遇到某一块蛋糕上草莓书为0，就跳出循环，草莓数最多n个，所以遍历块的话时间复杂度不会太高。详细看下面代码。

代码

void solve(){
    int W, H, n, a, b;
    cin >> W >> H >> n;
    vector<pii> cake;
    vector<int> A, B;
    
    for(int i = 0; i < n; i++) {
        int x, y; cin >> x >> y;
        cake.push_back({x, y});
    }
    cin >> a;
    for(int i = 0; i < a; i++) {
        int x; cin >> x;
        A.push_back(x);
    }
    A.push_back(1000000001);
    cin >> b;
    for(int i = 0; i < b; i++) {
        int x; cin >> x;
        B.push_back(x);
    }
    B.push_back(1000000001);
    sort(all(A)), sort(all(B));
 
    map<pii, int> mp;
    for(int i = 0; i < n; i++) {
        int l = lower_bound(all(A), cake[i].first) - A.begin();
        int r = lower_bound(all(B), cake[i].second) - B.begin();
        mp[{l, r}] ++; // 块(l,r)草莓数量+1
    }
 
    // ans
    int ans_max = 0, ans_min = 1e9;
    for(auto &[l, r]:mp) {
        ans_max = max(ans_max, r);
        ans_min = min(ans_min, r);
    }
    if(1ll * (a + 1) * (b + 1) > n) ans_min = 0;
    for(int i = 0; i <= a; i++) { // 时间复杂度O(n)的
        for(int j = 0; j <= b; j++) {
            ans_min = min(ans_min, mp[{i, j}]);
            if(ans_min == 0) break;
        }
    }
 
    cout << ans_min << ' ' << ans_max << '\n';
}

E - Good Graph

原题链接

E - Good Graph

题面

思路

题目给定的原图一定是不连通的，用并查集维护即可，对于k个条件，就是统计他们的祖先不能在同一个连通块中即可，处理q次询问即可。

代码

const int N = 2e5+100;
int p[N];
int find(int x) {
    return p[x] = (p[x] == x ? x : find(p[x]));
}
void merge(int u, int v) {
    u = find(u), v = find(v);
    if(u != v) p[u] = v;
}
set<int> st[N];
void solve(){
    int n, m; cin >> n >> m;
    for(int i = 1; i <= n; i++) p[i] = i;
    
    for(int i = 1; i <= m; i++) {
        int l, r; cin >> l >> r;
        merge(l, r);
    }
 
    int k; cin >> k;
    while(k --) {
        int u, v; cin >> u >> v;
        u = find(u), v = find(v);
        st[u].insert(v);
        st[v].insert(u);
    }
 
    int q; cin >> q;
    while(q --) {
        int u, v; cin >> u >> v;
        u = find(u), v = find(v);
        if(st[u].count(v) || st[v].count(u)) cout << "No\n";
        else cout << "Yes\n";
    }
}