Submission #22673

#TimeUsernameProblemLanguageResultExecution timeMemory
22673카시코이 (#40)Unifying Values (KRIII5_UV)C++14
7 / 7
156 ms4596 KiB
#include <cstdio> #include <cstring> #include <vector> #include <unordered_map> #include <algorithm> using namespace std; const int MOD = 1000000007; int n; long long arr[10004]; long long sum[10004]; long long allsum; int main() { //freopen("input.txt", "r", stdin); scanf("%d", &n); for (int i = 1; i <= n; ++i) { scanf("%lld", &arr[i]); sum[i] = sum[i - 1] + arr[i]; } allsum = sum[n]; long long ans = 0; if (allsum == 0) { int ppcnt = 0; long long ss = 0; for (int i = 1; i <= n; ++i) { ss += arr[i]; if (ss == 0) { ++ppcnt; } } ans = 1; for (int k = 1; k < ppcnt; ++k) { ans = (ans * 2ll) % MOD; } ans = (ans - 1 + MOD) % MOD; printf("%d\n", ans); return 0; } vector<int> cand; for (int i = 2; i <= n; ++i) { if (allsum % i == 0) { cand.push_back(i); } } vector<vector<int>> sum1(n + 1), prefix(n + 1); for (int cnt : cand) { long long targetsum = allsum / cnt; vector<vector<int>> idxs(n + 1); for (int i = 0; i <= n; ++i) { if (llabs(sum[i]) % llabs(targetsum) == 0) { long long mcnt = sum[i] / targetsum; if (0 <= mcnt && mcnt <= cnt) { idxs[mcnt].push_back(i); } } } idxs[0] = vector<int>{ 0 }; sum1[0] = vector<int>{ 1 }; prefix[0] = vector<int>{ 1 }; for (int k = 1; k <= cnt; ++k) { sum1[k].resize(idxs[k].size(), 0); prefix[k].resize(idxs[k].size(), 0); int p = 0; for (int j = 0; j < idxs[k].size(); ++j) { while (p < idxs[k - 1].size() && idxs[k - 1][p] < idxs[k][j]) { ++p; } sum1[k][j] = (p == 0) ? 0 : prefix[k - 1][p - 1]; prefix[k][j] = (j == 0 ? 0 : prefix[k][j - 1]) + sum1[k][j]; prefix[k][j] %= MOD; } } ans += sum1[cnt].back(); } printf("%d\n", ans % MOD); return 0; }

Compilation message (stderr)

UV.cpp: In function 'int main()':
UV.cpp:41:21: warning: format '%d' expects argument of type 'int', but argument 2 has type 'long long int' [-Wformat=]
   printf("%d\n", ans);
                     ^
UV.cpp:74:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for (int j = 0; j < idxs[k].size(); ++j) {
                      ^
UV.cpp:75:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     while (p < idxs[k - 1].size() && idxs[k - 1][p] < idxs[k][j]) {
              ^
UV.cpp:86:26: warning: format '%d' expects argument of type 'int', but argument 2 has type 'long long int' [-Wformat=]
  printf("%d\n", ans % MOD);
                          ^
UV.cpp:17:17: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d", &n);
                 ^
UV.cpp:19:25: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%lld", &arr[i]);
                         ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...