QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#320798#3850. DJ Darkoalgotester#WA 242ms66020kbC++146.3kb2024-02-03 21:32:022024-02-03 21:32:03

Judging History

This is the latest submission verdict.

  • [2024-02-03 21:32:03]
  • Judged
  • Verdict: WA
  • Time: 242ms
  • Memory: 66020kb
  • [2024-02-03 21:32:02]
  • Submitted

answer

#include "bits/stdc++.h"
using namespace std;

#define FOR(i,a,b) for (LL i = (a); i < (b); i++)
#define RFOR(i,b,a) for (LL i = (b) - 1; i >= (a); i--)
#define ITER(it,a) for (__typeof(a.begin()) it = a.begin(); it != a.end(); it++)
#define FILL(a,value) memset(a, value, sizeof(a))

#define SZ(a) (LL)a.size()
#define ALL(a) a.begin(), a.end()
#define PB push_back
#define MP make_pair

typedef long long LL;
typedef vector<LL> VI;
typedef pair<LL, LL> PII;

const double PI = acos(-1.0);
const LL INF = 1000 * 1000 * 1000 + 7;
const LL LINF = INF * (LL) INF;

const LL MAX = 1000200;
// const LL MAX = 100;


LL A[MAX];
LL B[MAX];
LL SB[MAX];

LL mrand()
{
	return abs((rand() << 16) ^ rand());
}
struct node
{
	LL l, r;	// children
	LL y;		// priority (should be random and different)
	LL cnt;	// size of subtree
	LL par;	// parent of the vertex
	LL val;	// value of the vertex
	LL push;	// reverse push

	LL ll;
	LL rr;
	
	void init(LL val, LL ll, LL rr) // init with value
	{
		l = r = -1;
		y = mrand();
		cnt = 1;
		par = -1;
		this->val = val;
		this->ll = ll;
		this->rr = rr;
		this->push = 0;
	}
};
// Minimum on subtree + reverse
struct Treap
{
	node A[MAX];
	LL sz;
	LL getCnt(LL x)
	{
		if (x == -1) return 0;
		return A[x].cnt;
	}
	LL newNode(LL val, LL ll, LL rr)
	{
		A[sz].init(val, ll, rr);
		sz++;
		return sz - 1;
	}
	LL PB(LL root, LL val, LL ll, LL rr)
	{
		return merge(root, newNode(val, ll, rr));
	}
	void upd(LL x)
	{
		if (x == -1) return;
		A[x].cnt = getCnt(A[x].l) + getCnt(A[x].r) + 1;
	}
	void add(LL x, LL val)
	{
		if (x == -1) return;
		A[x].val += val;
		A[x].push += val;
	}
	void push(LL x)
	{
		if (x == -1 || A[x].push == 0) return;
		add(A[x].l, A[x].push);
		add(A[x].r, A[x].push);
		A[x].push = 0;
	}
	PII split(LL x, LL cnt)
	{
		if (x == -1) return MP(-1, -1);
		if (cnt == 0) return MP(-1, x);
		push(x);
		LL left = getCnt(A[x].l);
		PII res;
		if (left >= cnt)
		{
			if (A[x].l != -1) A[A[x].l].par = -1;
			res = split(A[x].l, cnt);
			A[x].l = res.second;
			if (res.second != -1) A[res.second].par = x;
			res.second = x;
		}
		else
		{
			if (A[x].r != -1) A[A[x].r].par = -1;
			res = split(A[x].r, cnt - left - 1);
			A[x].r = res.first;
			if (res.first != -1) A[res.first].par = x;
			res.first = x;
		}
		upd(x);
		return res;
	}
	LL merge(LL x, LL y)
	{
		if (x == -1) return y;
		if (y == -1) return x;
		LL res;
		//if (mrand() % (getCnt(x) + getCnt(y)) < getCnt(x))
		if (A[x].y > A[y].y)
		{
			push(x);
			if (A[x].r != -1) A[A[x].r].par = -1;
			res = merge(A[x].r, y);
			A[x].r = res;
			if (res != -1) A[res].par = x;
			res = x;
		}
		else
		{
			push(y);
			if (A[y].l != -1) A[A[y].l].par = -1;
			res = merge(x, A[y].l);
			A[y].l = res;
			if (res != -1) A[res].par = y;
			res = y;
		}
		upd(res);
		return res;
	}

	LL getCntLeft(LL x)
	{
		LL prev = -1;
		LL res = 0;
		while(true)
		{
			if (prev != A[x].l) res += getCnt(A[x].l);
			LL par = A[x].par;
			if (par == -1) break;
			prev = x;
			if (A[par].r == x) res++;
			x = par;
		}

		return res;
	}

	LL dfs_find(LL x, LL pos)
	{
		// cout<<x<<' '<<pos<<": "<<A[x].ll<<' '<<A[x].rr<<' '<<A[x].l<<' '<<A[x].r<<endl;
		if (A[x].ll <= pos && pos <= A[x].rr) return x;
		if (pos < A[x].ll) return dfs_find(A[x].l, pos);
		return dfs_find(A[x].r, pos);
	}

	void get_v(LL x, vector<PII>& v)
	{
		if (x == -1) return;
		push(x);
		get_v(A[x].l, v);
		LL val = A[x].val;
		// LL cnt = A[x].rr - A[x].ll + 1;
		LL sum = SB[A[x].rr];
		if (A[x].ll != 0) sum -= SB[A[x].ll-1];
		v.PB(MP(val, sum));

		get_v(A[x].r, v);
	}

	void prLL(LL x)
	{
		if (x == -1) return;
		push(x);
		prLL(A[x].l);
		cout<<A[x].val<<" ("<<A[x].ll<<' '<<A[x].rr<<"), ";

		prLL(A[x].r);
	}
} T;


LL split_vertex_left(LL x, LL mid)
{
	LL y = T.newNode(T.A[x].val, T.A[x].ll, mid - 1);
	T.A[x].ll = mid;
	// cout<<"@@ "<<x<<": "<<T.A[x].ll<<' '<<T.A[x].rr<<endl;
	// cout<<"@@ "<<y<<": "<<T.A[y].ll<<' '<<T.A[y].rr<<endl;
	return y;
}

LL split_vertex_right(LL x, LL mid)
{
	// cout<<"# "<<x<<' '<<T.A[x].ll<<' '<<T.A[x].rr<<": "<<mid<<endl;
	LL y = T.newNode(T.A[x].val, mid + 1, T.A[x].rr);
	T.A[x].rr = mid;
	// cout<<"## "<<x<<": "<<T.A[x].ll<<' '<<T.A[x].rr<<endl;
	// cout<<"## "<<y<<": "<<T.A[y].ll<<' '<<T.A[y].rr<<endl;
	return y;
}

pair<LL, PII> getLR(LL root, LL l, LL r)
{
	LL x = T.dfs_find(root, l);
	LL y = T.dfs_find(root, r);
	// cout<<root<<' '<<l<<' '<<r<<endl;

	// cout<<"! "<<x<<' '<<y<<endl;


	PII a = T.split(root, T.getCntLeft(x));
	PII b = T.split(a.second, T.getCntLeft(y) + 1);

	root = b.first;

	if (T.A[x].ll != l)
	{
		LL v = split_vertex_left(x, l);
		a.first = T.merge(a.first, v);
	}

	if (T.A[y].rr != r)
	{
		LL v = split_vertex_right(y, r);
		b.second = T.merge(v, b.second);
	}

	return MP(a.first, MP(b.first, b.second));
}

LL root;

void update(LL l, LL r, LL x)
{
	auto R = getLR(root, l, r);
	LL mr = R.second.first;

	T.add(mr, x);

	root = T.merge(R.first, T.merge(R.second.first, R.second.second));
}

LL update(LL l, LL r)
{
	// cout<<"update "<<l<<' '<<r<<endl;
	auto R = getLR(root, l, r);

	// cout<<R.first<<' '<<R.second.first<<' '<<R.second.second<<endl;

	vector<PII> v;
	T.get_v(R.second.first, v);

	// FOR (i, 0, SZ(v))
	// {
	// 	cout<<v[i].first<<' '<<v[i].second<<", ";
	// }
	// cout<<endl;

	sort(ALL(v));
	LL s = 0;

	FOR (i, 0, SZ(v))
	{
		s += v[i].second;

	}

	LL c = 0;
	LL val = -1;
	FOR (i, 0, SZ(v))
	{
		c += v[i].second;
		if (c * 2 >= s)
		{
			val = v[i].first;
			break;
		}
	}

	LL x = T.newNode(val, l, r);

	root = T.merge(R.first, T.merge(x, R.second.second));

	return val;
}

int main(int argc, char* argv[])
{
	//ios::sync_with_stdio(false); cin.tie(0);
	// cout<<"&"<<endl;

	int n, q;
	scanf("%d%d", &n, &q);

	FOR (i, 0, n)
	{
		int x;
		scanf("%d", &x);
		A[i] = x;
	}

	FOR (i, 0, n)
	{
		int x;
		scanf("%d", &x);
		B[i] = x;
	}

	FOR (i, 0, n)
	{
		SB[i] = B[i];
		if (i != 0) SB[i] += SB[i-1];
	}

	// cout<<"*"<<endl;

	root = -1;
	FOR (i, 0, n)
	{
		root = T.PB(root, A[i], i, i);
	}

	FOR (i, 0, q)
	{
		// cout<<"===> ";
		// T.prLL(root);
		// cout<<endl;
		// cout<<"! "<<i<<endl;
		int t;
		scanf("%d", &t);
		if (t == 1)
		{
			int l, r, x;
			scanf("%d%d%d", &l, &r, &x);
			l--;
			r--;
			update(l, r, x);
		}
		if (t == 2)
		{
			int l, r;
			scanf("%d%d", &l, &r);
			l--;
			r--;
			LL res = update(l, r);
			printf("%lld\n", res);
		}
	}
	// cout<<"===> ";
	// T.prLL(root);
	// cout<<endl;

}

详细

Test #1:

score: 0
Wrong Answer
time: 242ms
memory: 66020kb

input:

200000 200000
185413631 745038744 881479208 394948467 101727403 796960399 284541402 80768155 286582974 546327406 197495962 552359542 727479505 437895671 143092948 7626834 741609268 540494577 298656274 548860413 41137417 210529949 658779847 161355446 486548926 602900245 119414972 310187428 238177860 ...

output:

462406736
1749348519
1749348519
467651597
874061694
874061694
874061694
874061694
-1521749
874061694
-893932302
-425504259
-425504259
332034115
332034115
86195778
86195778
86195778
86195778
86195778
-425504259
-165776398
-165776398
-165776398
-916781043
-254293799
-254293799
-254293799
-254293799
-8...

result:

wrong answer 93rd lines differ - expected: '-2052051040', found: '-1402522055'