AtCoder Beginner Contest 239 （C-E)

C - Knight Fork

题意: 给定两个点(x1, y1) (x2, y2), 询问是否存在一个点(x, y) (x y均为整数)使得这个点到这两个点的距离都是 根号5

思路：两个整数加起来等于5的方案有0 + 5， 1 + 4， 2 + 3三种方案，根据两点之间的距离公式可得只有1 + 4 = 1^2 + 2^2符合，因此我们可以用(x1, y1)来得到（x, y), 然后再用两点之间的距离公式来判断(x, y)到（x2, y2)的距离是否为 根号5 即可

代码:

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <cmath>

using namespace std;
typedef long long ll;
typedef pair<int, int> PII;

int main()
{    
    ll x1, y1, x2, y2;
    cin >> x1 >> y1 >> x2 >> y2;
    ll m, n;
    m = x1 - 1, n = y1 - 2;
    bool flag = false;
    if ((m-x2)*(m-x2) + (n-y2)*(n-y2)==5) flag = true;
    m = x1 + 1;
    if ((m-x2)*(m-x2) + (n-y2)*(n-y2)==5) flag = true;
    n = y1 + 2;
    if ((m-x2)*(m-x2) + (n-y2)*(n-y2)==5) flag = true;
    m = x1 - 1;
    if ((m-x2)*(m-x2) + (n-y2)*(n-y2)==5) flag = true;

    m = y1 - 1, n = x1 - 2;
    if ((n-x2)*(n-x2) + (m-y2)*(m-y2)==5) flag = true;
    n = x1 + 2;
    if ((n-x2)*(n-x2) + (m-y2)*(m-y2)==5) flag = true;
    m = y1 + 1;
    if ((n-x2)*(n-x2) + (m-y2)*(m-y2)==5) flag = true;
    n = x1 - 2;
    if ((n-x2)*(n-x2) + (m-y2)*(m-y2)==5) flag = true;

    
    if (flag) puts("Yes");
    else puts("No");
    return 0;
}

D - Prime Sum Game

题意：给定区间[A, B], [C, D], 询问是否能在[A, B]区间中选一个数x使得无论在[C, D]选任何一个数使得这两个数的和一定是合数，如果存在则输出Takahashi，否则输出Aoki

思路：观察到数据范围只有100，因此直接暴力就行

代码:

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

using namespace std;
typedef long long ll;
typedef pair<int, int> PII;

bool check(int x)
{
    if (x==2) return true;
    int i;
    for (i = 2; i * i <= x; i ++ )
        if (x % i == 0) break;
    return x % i != 0;
}

int main()
{    
    int a, b, c, d;
    cin >> a >> b >> c >> d;

    bool flag = false;
    for (int i = a; i <= b; i ++ )
    {
        bool s = true;
        for (int j = c; j <= d; j ++ )
            if (check(i+j)) s=false;
        if (s) flag = true;
    }
    if (flag) puts("Takahashi");
    else puts("Aoki");
    return 0;
}

E - Subtree K-th Max(树形dp）

题意：给定一颗根节点为1的树以及每个节点的权值，每次询问给出两个数v, k,即为询问v节点的子树的第K大的值是多少

思路：注意到这里的K最大也只有20(K很大的话就要用到主席树了)， 因此我们可以定义一个二维数组f[i][j]表示i节点的子树里面第j大的值。
对于从节点u向下走的边(u, v1), (u, v2), …, (u, vn),
那么f[u][j] 就等于 (f[v1][1], f[v1][2]…,f[v1][20], f[v2][1],f[v2][2],…,f[v2][20],f[vn][1],…,f[vn][20])中第j大的值，我们可以用优先队列来进行操作

代码：

#include <bits/stdc++.h>

using namespace std;
const int N = 1e5 + 10;
int w[N];
int f[N][30];
vector<int> ve[N];

void dfs(int x, int fa)
{
    priority_queue<int> q;
    q.push(w[x]);
    for (int i = 0; i < ve[x].size(); i ++ )
    {
        int y = ve[x][i];
        if (y == fa) continue;
        dfs(y, x);
        for (int j = 1; j <= 20; j ++ )
            q.push(f[y][j]);
    }

    for (int j = 1; j <= 20; j ++ )
    {
        if (q.empty()) break;
        f[x][j] = q.top();
        q.pop();
    }
}

int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ ) scanf("%d", &w[i]);
    for (int i = 1; i < n; i ++ )
    {
        int x, y;
        scanf("%d%d", &x ,&y);
        ve[x].push_back(y), ve[y].push_back(x);
    }

    dfs(1, -1);

    while (m -- )
    {
        int v, k;
        scanf("%d%d", &v, &k);
        printf("%d\n", f[v][k]);
    }
    return 0;
}

完结，如有错误还请指正，感谢！！！