7 files changed, 83 insertions, 0 deletions
diff --git a/contests/Round901/A/main.cpp b/contests/Round901/A/main.cpp
new file mode 100755
index 0000000..cd79e59
--- /dev/null
+++ b/contests/Round901/A/main.cpp
@@ -0,0 +1,68 @@
+#include <bits/stdc++.h>
+
+using namespace std;
+
+using ll = long long;
+using pii = pair<int, int>;
+using vpii = vector<pii>;
+using vi = vector<int>;
+using vll = vector<long long>;
+using mpii = map<int, int>;
+using mpll = map<ll, ll>;
+using db = long double;
+
+#define pb push_back
+#define all(x) (x).begin(), (x).end()
+#define rall(x) (x).rbegin(), (x).rend()
+#define lb lower_bound
+#define ub upper_bound
+#define make_unique(x)                                                         \
+  sort(all((x)));                                                              \
+  (x).resize(unique(all((x))) - (x).begin())
+#define ceil(a, b) ((a) + (b)-1) / (b)
+
+const int mod = (int)1e9 + 7;
+const db pi = acos((db)-1);
+const int dx[4]{1, 0, -1, 0};
+const int dy[4]{0, 1, 0, -1};
+
+#ifdef LOCAL
+#include "debug.h"
+#else
+#define dbg(...) 42
+#endif
+
+/* stuff you should look for:
+  ---------------------------
+   * special cases (n=1?)
+   * int overflow, array bounds
+   * do smth instead of nothing and stay organized
+   * WRITE STUFF DOWN
+   * DON'T GET STUCK ON ONE APPROACH
+ */
+
+void solve() {
+  int a, b, n;
+  cin >> a >> b >> n;
+  vi v(n);
+  for (int i = 0; i < n; i++) {
+    cin >> v[i];
+  }
+  int c = b;
+  ll ans = b - 1;
+  for (int i = 0; i < n; i++) {
+    c = min(a, 1 + v[i]);
+    ans += c - 1;
+  }
+  cout << ans + 1 << '\n';
+}
+
+int main() {
+  ios_base::sync_with_stdio(false);
+  cin.tie(NULL);
+  int tt;
+  cin >> tt;
+  while (tt--) {
+    solve();
+  }
+}
diff --git a/contests/Round901/A/main_input0.txt b/contests/Round901/A/main_input0.txt
new file mode 100644
index 0000000..d27f23e
--- /dev/null
+++ b/contests/Round901/A/main_input0.txt
@@ -0,0 +1,5 @@
+2
+5 3 3
+1 1 7
+7 1 5
+1 2 5 6 8
diff --git a/contests/Round901/A/main_input1.txt b/contests/Round901/A/main_input1.txt
new file mode 100644
index 0000000..23e1853
--- /dev/null
+++ b/contests/Round901/A/main_input1.txt
@@ -0,0 +1,3 @@
+1
+7 1 5
+1 5 2 8 6
+\ No newline at end of file
diff --git a/contests/Round901/A/main_input2.txt b/contests/Round901/A/main_input2.txt
new file mode 100644
index 0000000..aa2135a
--- /dev/null
+++ b/contests/Round901/A/main_input2.txt
@@ -0,0 +1,3 @@
+1
+5 3 3
+7 1 1
+\ No newline at end of file
diff --git a/contests/Round901/A/main_output0.txt b/contests/Round901/A/main_output0.txt
new file mode 100644
index 0000000..3a9b486
--- /dev/null
+++ b/contests/Round901/A/main_output0.txt
@@ -0,0 +1,2 @@
+9
+21
diff --git a/contests/Round901/A/main_output1.txt b/contests/Round901/A/main_output1.txt
new file mode 100644
index 0000000..b5045cc
--- /dev/null
+++ b/contests/Round901/A/main_output1.txt
@@ -0,0 +1 @@
+21
+\ No newline at end of file
diff --git a/contests/Round901/A/main_output2.txt b/contests/Round901/A/main_output2.txt
new file mode 100644
index 0000000..f11c82a
--- /dev/null
+++ b/contests/Round901/A/main_output2.txt
@@ -0,0 +1 @@
+9
+\ No newline at end of file