Submission #59515891

---

Source Code Expand

/*
 * Author: Number10
 * Created: 2024-11-07 20:11:10
 * Last Modified: 2024-11-07 20:23:18
 */

#include <bits/stdc++.h>

using namespace std;

const int N = 2e5 + 5;

int a[N], pre[N], nxt[N];
long long b[N];

int main() {
  int n;
  scanf("%d", &n);
  for (int i = 1; i <= n; ++i)
    scanf("%d", &a[i]);
  for (int i = 1; i <= n; ++i)
    pre[i] = pre[i - 1] + a[i];
  for (int i = n; i >= 1; --i)
    nxt[i] = nxt[i + 1] + a[i];

  long long sum = 0;
  for (int i = 1; i <= n; ++i) {
    b[i] = i;
    sum += i * a[i];
  }

  auto solve = [&]() -> bool {
    if (!sum) return true;
    int sig = sum < 0 ? -1 : 1;
    sum *= sig;
    int p = 0;
    for (int i = 1; i <= n; ++i)
      if (pre[i] == sig * 1) {
        p = i;
        break;
      }
    if (p) {
      for (int i = 1; i <= p; ++i)
        b[i] -= sum;
      return true;
    }
    for (int i = n; i >= 1; --i)
      if (nxt[i] == sig * -1) {
        p = i;
        break;
      }
    if (p) {
      for (int i = p; i <= n; ++i)
        b[i] += sum;
      return true;
    }
    return false;
  };
  if (!solve()) puts("No");
  else {
    puts("Yes");
    for (int i = 1; i <= n; ++i)
      printf("%lld ", b[i]);
  }
  return 0;
}

Submission Info

Compile Error

Main.cpp: In function ‘int main()’:
Main.cpp:18:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   18 |   scanf("%d", &n);
      |   ~~~~~^~~~~~~~~~
Main.cpp:20:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   20 |     scanf("%d", &a[i]);
      |     ~~~~~^~~~~~~~~~~~~

Judge Result