AtCoder Beginner Contest 262（A~D)

A. World Cup (循环判断）

判断一个数能否被4整除后余2，合法则输出，不合法则往后找

#include <bits/stdc++.h>
#define ll long long 
using namespace std;

int main()
{
    int n;
    cin >> n;

    while (n % 4 != 2) n++;

    cout << n << endl;

    return 0;
}

---

B. Triangle (Easier) (暴力)

给定一个无向图，判断图中有多少个环

#include <bits/stdc++.h>
#define ll long long 
using namespace std;
const int N = 110;

int n, m;
int g[N][N];

int main()
{
    cin >> n >> m;
    for (int i = 0; i < m; i++)
    {
        int u, v;
        cin >> u >> v;
        g[u][v] = g[v][u] = 1; //无向图标记
    }

    int res = 0; //三重循环判断是否存在闭环
    for (int i = 1; i <= n; i++)
        for (int j = i + 1; j <= n; j++)
            for (int k = j + 1; k <= n; k++)
                if (g[i][j] && g[i][k] && g[j][k])
                    res++;

    cout << res << endl;

    return 0;
}

---

C. Min Max Pair (分类讨论)

给定一个非负整数序列，查找满足以下所有条件的整数对 i, j
1 <= i < j <= N
min(a[i], a[j]) = i
max(a[i], a[j]) = j

满足的情况有两种：

① 如果 a[i] = j，a[j] = i，则 (i, j) 为一组答案

② 如果 a[i] = i 统计所有这样的点。假设有 n 个这样的点，那么答案就是从这 n 个点中任选 2 个，即 n *(n - 1) / 2.

最后因为 a[i] 有可能重复，所以需要记录下每个 a[i] 出现的位置

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 5e5 + 10;

int main()
{
    int n;
    cin >> n;

    vector<int> a(n + 1); //存储序列
    vector<int> b[n + 1]; //存储每个数出现的位置
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        b[a[i]].push_back(i);
    }

    ll res = 0, cnt = 0;
    for (int i = 1; i <= n; i++)
    {
        if (a[i] == i) cnt++; //第二种情况
        else {
            for (auto id : b[i])
                if (id == a[i])
                    res++; //第一种情况
        }
    }

    //因为 i,j 两个数为一组，所以cnt/2
    ll sum = res / 2 + (cnt * (cnt - 1)) / 2;

    cout << sum << endl;

    return 0;
}

---

D. I Hate Non-integer Number (动态规划)

给定一个序列，对于该序列的所有子集，判断 所有子集的平均值为整数 的子集有多少个
答案对 998244353 取模

动态规划：

用 j, k, l 来表示
f [j] [k] [l]表示前 j 个数中选 k 个数，其和% i 为 l 的 方案数
如果其平均值是整数，那么对 i 取模后结果必定为 0
即 f [n] [i] [0]为 所有 n 个数中任取 i 个数，其和的平均值为整数的方案数

得到状态转移方程：

不选 a[i] : f [j + 1] [k] [l] += f [j] [k] [l]

不选 a[i] : f [j + 1] [k + 1] [(l + a[j]) % i] += f [j] [k] [l]

#include <bits/stdc++.h>
#define ll long long 
#define mod 998244353
using namespace std;
const int N = 110;

ll a[N];
ll f[N][N][N];

int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
        cin >> a[i];

    ll res = 0;
    for (int i = 1; i <= n; i++) //%i意为着平均值对i取模
    {
        memset(f, 0, sizeof(f));
        f[0][0][0] = 1;
        for (int j = 0; j < n; j++)
            for (int k = 0; k <= n; k++) //选了i个数，循环k的上限应该为i，即k<=i,但后面有判断也没事
                for (int l = 0; l < i; l++)
                {
                    //不选第j个数，不选a[j]
                    f[j + 1][k][l] = (f[j + 1][k][l] + f[j][k][l]) % mod; //每次都需要取模
                    
                    //选第j个数，选a[j]
                    if (k < i)
                        f[j + 1][k + 1][(l + a[j]) % i] = (f[j + 1][k + 1][(l + a[j]) % i] + f[j][k][l]) % mod;
                }

        res = (res + f[n][i][0]) % mod;
    }

    cout << res << endl;

    return 0;
}

---