Update cowbasic.cpp

1 files changed, 2 insertions, 6 deletions

diff --git a/2017/US Open/Platinum/cowbasic.cpp b/2017/US Open/Platinum/cowbasic.cpp
index bf1343d..8ef44c9 100644
--- a/2017/US Open/Platinum/cowbasic.cpp
+++ b/2017/US Open/Platinum/cowbasic.cpp
@@ -38,15 +38,13 @@ void mult(matrix& A, matrix& B) { // B = A * B;
 	matrix res(d + 1, vi(d + 1, 0));
 	for (int i = 0; i <= d; i++) {
 		for (int j = 0; j <= d; j++) {
-			for (int k = 0; k <= d; k++) {
-				res[i][j] = ((ll)A[i][k] * (ll)B[k][j] + res[i][j]) % MOD;
-			}
+			for (int k = 0; k <= d; k++) res[i][j] = ((ll)A[i][k] * (ll)B[k][j] + res[i][j]) % MOD;
 		}
 	}
 	B = res;
 }
 
-void pow(matrix& A, int n) { // A ^ n
+void pow(matrix& A, int n) { // A = A ^ n
 	matrix res(d + 1, vi(d + 1, 0));
 	for (int i = 0; i <= d; i++) res[i][i] = 1;
 	while (n) {
@@ -62,13 +60,11 @@ int main() {
 
 	str input;
 	while (getline(cin, input)) program.push_back(input);
-
 	for (auto& line : program) { // Record variables
 		stringstream ss(line);
 		str n; ss >> n;
 		if (islower(n[0]) && var.find(n) == var.end()) var[n] = d++;
 	}
-
 	mat_st.push_back({});
 	mat_st.back().resize(d + 1, vi(d + 1, 0));
 	for (int i = 0; i <= d; i++) mat_st.back()[i][i] = 1;