1927C - Choose the Different Ones - CodeForces Solution


brute force greedy math

Please click on ads to support us..

C++ Code:

#include <bits/stdc++.h>
using namespace std;
#define ll long long
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);

    int t; cin >> t;
    while(t--){
        int n,m,k; cin >> n >> m >> k;
        int a[k+1], b[k+1];
        fill(a,a+k+1,0);
        fill(b,b+k+1,0);
        for(int i = 0; i < n; i++){
            int temp; cin >> temp;
            if(temp <= k){
                a[temp] = 1;
            }      
        }
        for(int i = 0; i < m; i++){
            int temp; cin >> temp;
            if(temp <= k){
                b[temp] = 1;
            }    
        }
        int ahave = 0, bhave = 0, ans = 1;
        for(int i = 1; i <= k; i++){
            if(a[i] == 0 and b[i] == 1){
                bhave += 1;
            } else if(a[i] == 1 and b[i] == 0){
                ahave += 1;
            } else if(a[i] == 0 and b[i] == 0){
                ans = 0;
            }
        }

        if(ahave > k/2 or bhave > k/2){
            ans = 0;
        }
        if(ans == 0){
            cout << "NO";
        } else {
            cout << "YES";
        }
        cout << "\n";
    }
    
    return 0;    
}


Comments

Submit
0 Comments
More Questions

1302. Deepest Leaves Sum
1209. Remove All Adjacent Duplicates in String II
994. Rotting Oranges
983. Minimum Cost For Tickets
973. K Closest Points to Origin
969. Pancake Sorting
967. Numbers With Same Consecutive Differences
957. Prison Cells After N Days
946. Validate Stack Sequences
921. Minimum Add to Make Parentheses Valid
881. Boats to Save People
497. Random Point in Non-overlapping Rectangles
528. Random Pick with Weight
470. Implement Rand10() Using Rand7()
866. Prime Palindrome
1516A - Tit for Tat
622. Design Circular Queue
814. Binary Tree Pruning
791. Custom Sort String
787. Cheapest Flights Within K Stops
779. K-th Symbol in Grammar
701. Insert into a Binary Search Tree
429. N-ary Tree Level Order Traversal
739. Daily Temperatures
647. Palindromic Substrings
583. Delete Operation for Two Strings
518. Coin Change 2
516. Longest Palindromic Subsequence
468. Validate IP Address
450. Delete Node in a BST