Submission #994428

---

Source Code Expand

#include <cstdio>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <cstring>
#include <string>
#include <cmath>
#include <cstdlib>
#include <iostream>

using namespace std;

#define MOD @
#define ADD(X,Y) ((X) = ((X) + (Y)) % MOD)
typedef long long i64; typedef vector<int> ivec; typedef vector<string> svec;

int N, A[202020], M, X[202020];
i64 dp[202020];
i64 S[202020];

int uf[4040404];

int root(int p)
{
	return uf[p] < 0 ? p : (uf[p] = root(uf[p]));
}
void join(int p, int q)
{
	p = root(p);
	q = root(q);
	if (p == q) return;

	if (p < q) swap(p, q);
	uf[p] += uf[q];
	uf[q] = p;
}

const int OFS = 2010101;

void precalc()
{
	S[0] = 0;
	for (int i = 1; i < N; ++i) S[i] = S[i - 1] + A[i - 1];
	dp[N - 1] = 0;
//	i64 acc = A[N - 1] - dp[N - 1] - S[N - 1];
	for (int i = 0; i <= 2 * OFS; ++i) uf[i] = -1;

	vector<int> unres;
	for (int i = OFS; i >= -OFS; --i) unres.push_back(i);

	for (int i = N - 2; i >= 2; --i) {
		// acc = max(acc, A[i] - (acc + S[i + 1]) - S[i])
		int app = A[i] - S[i + 1] - S[i];
		while (!unres.empty()) {
			int v = unres[unres.size() - 1];
			if (v < app - v) {
				join(v + OFS, app - v + OFS);
				unres.pop_back();
			} else {
				break;
			}
		}

	}
}

i64 solve()
{
	i64 acc = A[N - 1] - S[N - 1];
	if (acc > OFS) {
	} else if (acc < -OFS) {
		acc = -acc;
	} else {
		acc = root(acc + OFS) - OFS;
	}
	/*
	for (int i = N - 2; i >= 2; --i) {
		acc = max(acc, A[i] - S[i + 1] - S[i] - acc);
	}
	*/

	for (int i = 1; i >= 1; --i) {
		dp[i] = acc + S[i + 1];
		acc = max(acc, A[i] - dp[i] - S[i]);
	}
	return dp[1] + A[0] - A[1];
}

int main()
{
	scanf("%d", &N);
	for (int i = 0; i < N - 1; ++i) scanf("%d", A + i);
	scanf("%d", &M);
	for (int i = 0; i < M; ++i) scanf("%d", X + i);

	precalc();
	for (int i = 0; i < M; ++i) {
		A[N - 1] = X[i];
		printf("%lld\n", solve());
	}
	return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:94:17: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d", &N);
                 ^
./Main.cpp:95:52: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  for (int i = 0; i < N - 1; ++i) scanf("%d", A + i);
                                                    ^
./Main.cpp:96:17: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d", &M);
                 ^
./Main.cpp:97:48: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  for (int i = 0; i < M; ++i) scanf("%d", X + i);
                                                ^

Judge Result