Educational Codeforces Round 115 (Rated for Div. 2)

B. Groups

思路：

只要找到是否有兩天滿足條件即可，我們可以這么分析，對於任意的兩天，看這n組學生：

一天有課且另一天沒課的記為cnt1

一天沒課且另一天有課的記為cnt2

兩天都有課的記為cnt3

兩天都沒課的記為cnt4

而兩天都有課cnt3的可以放到cnt1中也可以放到cnt2中，我們只要滿足cnt1+cnt3>=n/2，且cnt2+cnt3>=n/2，且cnt4==0，就代表有方案

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
#include <set>
#include <vector>

#define x first
#define y second

using namespace std;

typedef long long LL;
typedef pair<int, int>PII;

const int N = 200010;
const int MOD = 1000000007;

int a[N][6];

int main()
{
    int T;
    cin >> T;
    while (T--)
    {
        int n;
        cin >> n;
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= 5; j++)
                cin >> a[i][j];

        bool flag = false;
        for (int i = 1; i <= 5; i++)
            for (int j = i + 1; j <= 5; j++)
            {
                int cnt1 = 0, cnt2 = 0, cnt3 = 0, cnt4 = 0;
                for (int k = 1; k <= n; k++)
                {
                    if (a[k][i] && !a[k][j]) cnt1++;
                    if (!a[k][i] && a[k][j]) cnt2++;
                    if (a[k][i] && a[k][j])  cnt3++;
                    if (!a[k][i] && !a[k][j]) cnt4++;
                }
                if (cnt1 + cnt3 >= n / 2 && cnt2 + cnt3 >= n / 2 && cnt4 == 0) flag = true;
            }

        if (flag) cout << "YES" << endl;
        else cout << "NO" << endl;
    }
    return 0;
}

 D. Training Session

思路：

排列組合問題，容斥原理，首先正難則反，我們可以先求出全集，然后減去不符合條件的個數，假設有A,B,C三個人，那么不符合的情況就為A的a屬性和B相同，且A的b屬性和C相同，就像下面這種情況：

 

那么當前這個數不滿足條件的總數就為（和A相同的個數-1）乘上（和B相同個數的-1），-1是減去本身

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <set>
#include <vector>
#include <map>
#include <unordered_set>
#include <unordered_map>

#define x first
#define y second
#define IOS ios::sync_with_stdio(false);cin.tie(0);

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;

const int N = 200010, M = 100010, MOD = 1000000007, INF = 0x3f3f3f3f;

int a[N], b[N];
map<int, int> mpa, mpb;

int main()
{
    IOS;
    int T;
    cin >> T;
    while(T -- )
    {
        int n;
        cin >> n;
        mpa.clear();
        mpb.clear();
        for(int i = 1; i <= n; i ++ ) 
        {
            cin >> a[i] >> b[i];
            mpa[a[i]]++, mpb[b[i]]++;
        }
        LL res = (LL)n * (n - 1) * (n - 2) / 6;
        for (int i = 1; i <= n; i ++ )
            res -= (LL)(mpa[a[i]] - 1) * (mpb[b[i]] - 1);

        cout << res << endl;
    }
    return 0;
}