VP TOYOTA SYSTEMS Programming Contest 2024（AtCoder Beginner Contest 377)

---

A - Rearranging ABC

点击查看代码

void solve() {
    int cnt[26]{};
    std::string s;
    std::cin >> s;
    for (auto & c : s) {
    	++ cnt[c - 'A'];
    }

    if (cnt[0] == cnt[1] && cnt[1] == cnt[2] && cnt[2] == 1) {
    	std::cout << "Yes\n";
    } else {
    	std::cout << "No\n";
    }
}

---

B - Avoid Rook Attack

题意：棋盘上有$n$个棋子，每个棋子都能吃到所在行列的格子。问有多少格子不会被吃。

记录被吃的行和列。一个格子不会被吃那么它所在的行列都不会被吃。

点击查看代码

void solve() {
	int n = 8;
    std::vector<std::string> s(n);
    for (int i = 0; i < n; ++ i) {
    	std::cin >> s[i];
    }

    std::vector<int> row(n), col(n);
    for (int i = 0; i < n; ++ i) {
    	for (int j = 0; j < n; ++ j) {
    		if (s[i][j] == '#') {
    			row[i] = col[j] = 1;
    		}
    	}
    }

    int ans = 0;
    for (int i = 0; i < n; ++ i) {
    	for (int j = 0; j < n; ++ j) {
    		ans += row[i] == 0 && col[j] == 0;
    	}
    }

    std::cout << ans << "\n";
}

---

C - Avoid Knight Attack

题意：棋盘上有$n$个棋子，棋子是象棋里马的走法。问有多少格子不会被吃。

把被吃的格子用$set$存下来。

点击查看代码

void solve() {
	int n, m;
	std::cin >> n >> m;
	const int dx[] = {-2, -1, 1, 2, 2, 1, -1, -2}, dy[] = {1, 2, 2, 1, -1, -2, -2, -1};
	std::set<std::pair<int, int>> s;
	for (int i = 0; i < m; ++ i) {
		int x, y;
		std::cin >> x >> y;
		-- x, -- y;
		s.insert({x, y});
		for (int j = 0; j < 8; ++ j) {
			int nx = x + dx[j], ny = y + dy[j];
			if (nx < 0 || nx >= n || ny < 0 || ny >= n) {
				continue;
			}

			s.insert({nx, ny});
		}
	}    

	std::cout << (i64)n * n - (int)s.size() << "\n";
}

---

D - Many Segments 2

题意：有$n$个区间，求有多少区间不包含其中任意一个区间。值域为$[1,m]$。

把每个右端点对应的左端点存下来。然后枚举右端点，这个右端点可以往左取到的地方就是前面所有区间的$l$的最大值。

点击查看代码

void solve() {
    int n, m;
    std::cin >> n >> m;
    std::vector<std::vector<int>> R(m + 1);
    for (int i = 0; i < n; ++ i) {
    	int l, r;
    	std::cin >> l >> r;
    	R[r].push_back(l);
    }

    i64 ans = 0;
    int l = 0;
    for (int r = 1; r <= m; ++ r) {
    	for (auto & p : R[r]) {
    		l = std::max(l, p);
    	}

    	ans += r - l;
    }

    std::cout << ans << "\n";
}

---

E - Permute K times 2

题意：给你一个排列，每次使得$p_i=p_{p_i}$。问进行$k$次后的排列是什么。

如果按$i$向$p_i$建边，那么因为是一个排列，每个点恰好有一个出度一个入度，就会形成若干个环。这些环不管操作几次都是这些数。于是我们观察每一个环的变化。跳一次变成$p_{p_i}$，二次变成$p_{p_{p_{p_i}}}$，那么发现每个点走了$2^k$次。

点击查看代码

void solve() {
    int n;
    i64 k;
    std::cin >> n >> k;
    std::vector<int> p(n);
    for (int i = 0; i < n; ++ i) {
    	std::cin >> p[i];
    	-- p[i];
    }

    auto power = [&](int a, i64 b, int m) -> int {
    	int res = 1;
    	for (; b; b >>= 1, a = (i64)a * a % m) {
    		if (b & 1) {
    			res = (i64)res * a % m;
    		}
    	}

    	return res;
    };

    std::vector<int> st(n), ans(n);
    for (int i = 0; i < n; ++ i) {
    	if (!st[i]) {
    		int j = i;
    		std::vector<int> a;
    		while (!st[j]) {
    			st[j] = 1;
    			a.push_back(j);
    			j = p[j];
    		}

    		int m = a.size();
    		int d = power(2, k, m);	

    		for (int i = 0; i < m; ++ i) {
    			ans[a[i]] = p[a[(i + d - 1 + m) % m]];
    		}
    	}
    }

    for (int i = 0; i < n; ++ i) {
    	std::cout << ans[i] + 1 << " \n"[i == n - 1];
    }
}

---

G - Edit to Match

题意：依次输入$n$个字符串，对于第$i$个字符串，你可以每次花1的代价删除最后一个字符，或者在最后面加一个字符。问让第$i$个字符串和前面某一个字符串相等的最少代价。

建立一棵字典树，对于字典树的每一个节点$p$，记$f[p]$为前缀匹配到了$p$，还需要往后面加多少字符串可以和已有的字符串相等的最小代价。那么每次插入时到一个节点就看要删几个字符到这个前缀，然后加上$f[p]$。同时把插入时遇到的$p$存下来，然后从后往前更新。

点击查看代码

const int TrieN = 1e6 + 5;
int trie[TrieN][26], f[TrieN * 26];
struct Trie {
	int idx;
	Trie() {
		idx = 0;
	}

	int newNode() {
		idx += 1;
		memset(trie[idx], 0, sizeof trie[idx]);
		f[idx] = 1e9;
		return idx;
	}

	int insert(const std::string & s) {
		int p = 0;
		std::vector<int> a;
		int res = 1e9;
		int len = s.size();
		for (auto & c : s) {
			int x = c - 'a';
			if (!trie[p][x]) {
				trie[p][x] = newNode();
			}

			p = trie[p][x];
			-- len;
			res = std::min(res, f[p] + len);
			a.push_back(p);
		}

		int n = a.size();

		f[a[n - 1]] = 0;
		for (int i = n - 2; i >= 0; -- i) {
			f[a[i]] = std::min(f[a[i]], f[a[i + 1]] + 1);
		}

		return res;
	}
};

void solve() {
    int n;
    std::cin >> n;
    Trie tr;
    for (int i = 0; i < n; ++ i) {
    	std::string s;
    	std::cin >> s;
    	std::cout << std::min((int)s.size(), tr.insert(s)) << "\n";
    }
}