usaco

Clean implementations of solutions to USACO problems

  1. 1
  2. 2
  3. 3
  4. 4
  5. 5
  6. 6
  7. 7
  8. 8
  9. 9
  10. 10
  11. 11
  12. 12
  13. 13
  14. 14
  15. 15
  16. 16
  17. 17
  18. 18
  19. 19
  20. 20
  21. 21
  22. 22
  23. 23
  24. 24
  25. 25
  26. 26
  27. 27
  28. 28
  29. 29
  30. 30
  31. 31
  32. 32
  33. 33
  34. 34
  35. 35
  36. 36
  37. 37
  38. 38
  39. 39
  40. 40
  41. 41
  42. 42
  43. 43
  44. 44
  45. 45
  46. 46
  47. 47
  48. 48
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

ll a[30][30], cost[100001][30] = { 0 }, DP[100001][30];

int main() {
	ifstream cin("cowmbat.in");
	ofstream cout("cowmbat.out");

	int N, M, K;
	string S;
	cin >> N >> M >> K >> S;
	for (int i = 0; i <= 26; ++i) {
		for (int j = 0; j <= 26; ++j) a[i][j] = 1e9;
	}
	for (int i = 0; i < M; ++i) {
		for (int j = 0; j < M; ++j) cin >> a[i][j];
	}

	for (int k = 0; k < M; ++k) {
		for (int i = 0; i < M; ++i) {
			for (int j = 0; j < M; ++j) a[i][j] = min(a[i][k] + a[k][j], a[i][j]);
		}
	}

	for (int i = 0; i < K; ++i) {
		for (int j = 0; j <= 26; ++j) cost[0][j] += a[S[i] - 'a'][j];
	}
	for (int i = 0; i < N - K; ++i) {
		for (int j = 0; j <= 26; ++j) {
			cost[i + 1][j] = a[S[i + K] - 'a'][j] - a[S[i] - 'a'][j] + cost[i][j];
		}
	}

	for (int i = 0; i <= N; ++i) {
		for (int j = 0; j <= 26; ++j) DP[i][j] = 1e9;
	}
	DP[0][26] = 0;
	for (int i = 0; i < N; ++i) {
		for (int j = 0; j <= 26; ++j) DP[i + 1][j] = min(a[S[i] - 'a'][j] + DP[i][j], DP[i + 1][j]);
		int best = *min_element(DP[i], DP[i] + 27);
		if (i + K <= N) {
			for (int j = 0; j <= 26; ++j) DP[i + K][j] = min(best + cost[i][j], DP[i + K][j]);
		}
	}
	cout << *min_element(DP[N], DP[N] + 27);
}