D. Exam in MAC - 思维

题面

分析

可以计算出 $0$ 到 $c$ 之间所有的对数，一共有 $c+1$ 个数，所以有(c + 1) * (c + 2) / 2 对。然后考虑什么情况可以导致出现不符合的情况：

• 1.可能会存在 $x+y$ 在集合中出现过的，那么对于集合中每一个元素 $s_i$ ，可以得出对于 $0\le x\le$ $\frac{s_i}{2}$$+1$，每一个数都有另一个范围内的数存在对应，因此这部分都要减去。
• 2.可能存在 $y-x$ 在集合中出现过的， 那么对于集合中每一个元素 $s_i$ ，可以得出对于 $s_i\le y\le c$，每一个数都有另一个范围内的数存在对应，因此这部分都要减去。
• 3.以上两部分减去了一部分重复的，也就是 $x+y$ 和 $y-x$ 同时在集合中的，所以要将这两部分加起来，设 $x+y=s_i$, $y-x=s_j$，那么解方程可以得到 $x$ = $\frac{s_i-s_j}{2}$, $y$ = $\frac{s_i+s_j}{2}$，那么如果 $s_i和s_j$的奇偶性不一样，那么$x和y$就不是整数，说明不存在，其他情况存在。

代码

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

void solve() {
    int n;
    ll c;
    cin >> n >> c;
    vector<ll> x(n);
    for(int i = 0; i < n; i ++) cin >> x[i];
    ll ans = (c + 1) * (c + 2) / 2;
    int cnt1 = 0, cnt2 = 0;
    for(int i = 0; i < n; i ++) {
        ans -= x[i] / 2 + 1;
        ans -= c - x[i] + 1;
        if(x[i] & 1) cnt1 ++;
        else cnt2 ++;
    }
    ans += (ll)cnt1 * (cnt1 + 1) / 2;
    ans += (ll)cnt2 * (cnt2 + 1) / 2;
    cout << ans << "\n";
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T;
    cin >> T;
    while(T --) {
        solve();
    }
}