QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#667271#9492. 树上简单求和Galex19 4617ms90884kbC++236.0kb2024-10-22 21:59:372024-10-22 21:59:44

Judging History

This is the latest submission verdict.

  • [2024-10-22 21:59:44]
  • Judged
  • Verdict: 19
  • Time: 4617ms
  • Memory: 90884kb
  • [2024-10-22 21:59:37]
  • Submitted

answer

#include <bits/stdc++.h>
#define LL long long
#define ULL unsigned long long
#define pii pair<int, int>
#define pll pair<LL, LL>
#define fi first
#define se second
#define mkp make_pair
#define sz(x) (int)x.size()
using namespace std;
bool Mst;

LL read() {
	LL s = 0, f = 1;
	char ch = getchar();
	while (ch < '0' || ch > '9')
		f = (ch == '-' ? -1 : 1), ch = getchar();
	while (ch >= '0' && ch <= '9')
		s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar();
	return s * f;
}

ULL readull() {
	ULL s = 0; char ch = getchar();
	while (ch < '0' || ch > '9') ch = getchar();
	while (ch >= '0' && ch <= '9') s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar();
	return s;
}

mt19937 rnd(1919810);

const int MAXN = 200005;

int n, m;
ULL a[MAXN], suma[MAXN];
vector<int> e[MAXN];
int fa[MAXN][20], dep[MAXN], Fa[MAXN];
int dfn[MAXN], tim = 0, rnk[MAXN];
int o[MAXN], id[MAXN];

void pre(int x, int fat) {
	fa[x][0] = Fa[x] = fat, dep[x] = dep[fat] + 1;
	dfn[x] = ++tim, rnk[tim] = x, suma[x] = suma[fat] + a[x];
	for (int y : e[x])
		if (y != fat)
			pre(y, x);
}

int mov(int u, int d) {
	for (int i = 18; i >= 0; i--)
		if ((1 << i) <= d)
			d -= (1 << i), u = fa[u][i];
	return u;
}

int lca(int u, int v) {
	dep[u] > dep[v] ? u = mov(u, dep[u] - dep[v]) : v = mov(v, dep[v] - dep[u]);
	if (u == v) return u;
	for (int i = 18; i >= 0; i--)
		if (fa[u][i] != fa[v][i])
			u = fa[u][i], v = fa[v][i];
	return fa[u][0];
}

namespace Blocker {
	const int MAXC = 505;
	int B, C;
	int L[MAXC], R[MAXC], bl[MAXN];
	ULL sum[MAXC], val[MAXN];
	void init() {
		B = sqrt(n), C = (n - 1) / B + 1;
		for (int i = 1; i <= n; i++)
			bl[i] = (i - 1) / B + 1, R[bl[i]] = i;
		for (int i = n; i; i--)
			L[bl[i]] = i;
	}
	void mdf(int x, ULL v) {
		for (int i = x; i <= R[bl[x]]; i++) val[i] += v;
		for (int i = bl[x]; i <= C; i++) sum[i] += v;
	}
	inline ULL qry(int x) { return val[x] + sum[bl[x] - 1];}
	inline ULL qry(int l, int r) {
		return qry(r) - qry(l - 1);
	}
}

namespace Tree1 {
	vector<int> e[MAXN];
	void add(int u, int v) {
		e[u].push_back(v);
		e[v].push_back(u);
	}
	int fa[MAXN][20], dep[MAXN], dfn[MAXN], rnk[MAXN], tim = 0, sz[MAXN], o[MAXN], Fa[MAXN];
	void pre(int x, int fat) {
		dfn[x] = ++tim, rnk[tim] = x, sz[x] = 1;
		fa[x][0] = Fa[x] = fat, dep[x] = dep[fat] + 1;
		for (int y : e[x])
			if (y != fat)
				pre(y, x), sz[x] += sz[y];
	}
	int mov(int u, int d) {
		for (int i = 18; i >= 0; i--)
			if ((1 << i) <= d)
				d -= (1 << i), u = fa[u][i];
		return u;
	}
	int lca(int u, int v) {
		dep[u] > dep[v] ? u = mov(u, dep[u] - dep[v]) : v = mov(v, dep[v] - dep[u]);
		if (u == v) return u;
		for (int i = 18; i >= 0; i--)
			if (fa[u][i] != fa[v][i])
				u = fa[u][i], v = fa[v][i];
		return fa[u][0];
	}
	void pre() {
		pre(1, 0);
		for (int j = 1; j <= 18; j++)
			for (int i = 1; i <= n; i++)
				fa[i][j] = fa[fa[i][j - 1]][j - 1];
		for (int i = 1; i <= n; i++) o[i] = i;
		sort(o + 1, o + n + 1, [](int x, int y) {return dep[x] < dep[y];});
	}
	void mdf(int x, int y, int l, ULL k) {
		Blocker::mdf(dfn[x], k), Blocker::mdf(dfn[y], k);
		Blocker::mdf(dfn[l], -k), Blocker::mdf(dfn[Fa[l]], -k);
	}
	inline ULL qry(int x) {
		return Blocker::qry(dfn[x], dfn[x] + sz[x] - 1) + a[x];
	}
	int sum[MAXN];
	void intot() {
		for (int i = 2; i <= n; i++) {
			int x = o[i];
			sum[x] += sum[Fa[x]];
		}
	}
	int qry(int u, int v, int l) {
		return sum[u] + sum[v] - sum[l] - sum[Fa[l]];
	}
}

int p[MAXN], cnt = 0;
bitset<MAXN> in;

void spread() {
	cnt = 1500;
	for (int i = 1; i <= cnt; i++) p[i] = rnd() % n + 1;
	p[++cnt] = 1;
	sort(p + 1, p + cnt + 1, [](int x, int y) {return dfn[x] < dfn[y];});
	for (int i = 2, len = cnt; i <= len; i++) p[++cnt] = lca(p[i], p[i - 1]);
	sort(p + 1, p + cnt + 1, [](int x, int y) {return dfn[x] < dfn[y];});
	cnt = unique(p + 1, p + cnt + 1) - p - 1;
	int mx = 0;
	for (int i = 2; i <= cnt; i++) mx = max(mx, dep[p[i]] - dep[p[i - 1]]);
	cerr << mx << endl;
	for (int i = 1; i <= cnt; i++) in[p[i]] = 1;
}

int x[MAXN], y[MAXN], l1[MAXN];
int p1[MAXN], p2[MAXN], p3[MAXN];
ULL ans[MAXN], k[MAXN];
int MAX = 0, CNT = 0;

int qry(int u, int i, ULL v) {
	while (!in[u])
		ans[i] += v * Tree1::qry(u), u = Fa[u], CNT++;
	return u;
}

bool Med;
signed main() {
	// freopen("mist5-1.in", "r", stdin);
	// freopen("1.out", "w", stdout);
	fprintf(stderr, "%.2lfMB\n", abs(&Med - &Mst) / 1048576.0);
	n = read(), m = read();
	for (int i = 1; i <= n; i++) a[i] = readull();
	for (int i = 1; i < n; i++) {
		int u = read(), v = read();
		Tree1::add(u, v);
	}
	for (int i = 1; i < n; i++) {
		int u = read(), v = read();
		e[u].push_back(v), e[v].push_back(u);
	}
	pre(1, 0), Tree1::pre();
	for (int j = 1; j <= 18; j++)
		for (int i = 1; i <= n; i++)
			fa[i][j] = fa[fa[i][j - 1]][j - 1];
	Blocker::init(), spread();
	cerr << cnt << endl;
	for (int i = 1; i <= m; i++) {
		x[i] = read(), y[i] = read(), k[i] = readull(), l1[i] = Tree1::lca(x[i], y[i]);
		Tree1::mdf(x[i], y[i], l1[i], k[i]);
		p1[i] = qry(x[i], i, 1), p2[i] = qry(y[i], i, 1);
		int l = lca(x[i], y[i]);
		p3[i] = qry(l, i, -2), ans[i] += Tree1::qry(l);
	}
	cerr << CNT << endl;
	cerr << 1.0 * clock() / CLOCKS_PER_SEC << endl;
	for (int i = 1; i <= n; i++) o[i] = i;
	sort(o + 1, o + n + 1, [](int x, int y) {return dep[x] < dep[y];});
	for (int i = 1; i <= n; i++) id[o[i]] = i;
	cerr << 1.0 * clock() / CLOCKS_PER_SEC << endl;
	for (int i = 1; i <= cnt; i++) {
		memset(Tree1::sum, 0, sizeof Tree1::sum);
		ULL res = suma[p[i]];
		Tree1::sum[p[i]]++;
		for (int j = id[p[i]]; j > 1; j--) {
			int x = o[j];
			Tree1::sum[Fa[x]]++;
		}
		Tree1::intot();
		for (int j = 1; j <= m; j++) {
			int c = Tree1::qry(x[j], y[j], l1[j]);
			// cout << c << endl;
			res += k[j] * c;
			if (p1[j] == p[i]) ans[j] += res;
			if (p2[j] == p[i]) ans[j] += res;
			if (p3[j] == p[i]) ans[j] -= res * 2;
		}
	}
	for (int i = 1; i <= m; i++)
		printf("%llu\n", ans[i]);
	cerr << 1.0 * clock() / CLOCKS_PER_SEC << endl;
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Subtask #1:

score: 0
Wrong Answer

Test #1:

score: 0
Wrong Answer
time: 34ms
memory: 22816kb

input:

3000 3000
7236742292501328495 17973811477309806363 16075782662531676171 17971236571771878676 11392080645527132110 3685563455925680459 9773593720088356683 8313828403245053795 7736401634567449043 1634817828009987181 6951124933529719486 12775126714635387213 15460977209223753216 397573676785925632 31372...

output:

6632864750723193072
6617659476937272874
9761493200078153761
6909778739567197619
5493858549609581927
13397436909123390273
8564106543213223133
8532217533728527311
1630065571937948158
6157454189936495313
6743904859699756081
12862060668458629433
2358643317263130267
12907213503814510416
28130593481371249...

result:

wrong answer 1st lines differ - expected: '12105153858659381124', found: '6632864750723193072'

Subtask #2:

score: 0
Skipped

Dependency #1:

0%

Subtask #3:

score: 0
Skipped

Dependency #2:

0%

Subtask #4:

score: 0
Wrong Answer

Test #21:

score: 0
Wrong Answer
time: 3160ms
memory: 81964kb

input:

200000 200000
622783158027686223 2242697872372232537 8481648430436878777 10092474834140799044 15403999682625301609 12614289513474949582 9180944589267018841 7823784919308285798 8257785171198951273 5134508521895120821 8041682272181381093 3835432206618893170 2653803171409877650 5589823419153460372 1007...

output:

7385470938339111187
1015969055890815299
2272020447389508254
17954309321915788568
13260662887021152030
1851369439765398725
5354567715568928238
9833409675563275379
8595022223193976663
6143196964833666641
7528168595372072271
13360958084876033860
5758376512484743275
10395084829076379515
6067863346058934...

result:

wrong answer 1st lines differ - expected: '9042998055336671259', found: '7385470938339111187'

Subtask #5:

score: 0
Wrong Answer

Test #27:

score: 0
Wrong Answer
time: 4617ms
memory: 90140kb

input:

200000 200000
1958469220619413759 14991498002015735322 6054491201406941902 18206143187746582567 15082377615826460430 2936248617457291604 10073577150351675920 16534472678586906457 2207599132486246393 10301540360769075442 1492580560381080472 551692353431379140 13238280352539145808 8462626987240986565 ...

output:

17609044785804256917
10888391581081477040
7664363696211351499
5480821562831889252
13868651397914544853
17670110099396310117
1814039980563878003
1717536792026800005
17312050420753525411
742425543325461764
8113962056160164434
1011923303415334401
1704749290057566134
11613220426756932450
720531681472452...

result:

wrong answer 1st lines differ - expected: '11479812171669345085', found: '17609044785804256917'

Subtask #6:

score: 19
Accepted

Test #34:

score: 19
Accepted
time: 2338ms
memory: 90796kb

input:

200000 200000
6794776813641982926 1561596256197101737 10910039723053043515 7892247858295192798 12233819960547881004 17695389034783066733 9173201689566865598 17626618141377486739 7358781671024283919 6787559733384974662 3884392438269280436 14872846228351316833 9037842441501571648 14299818404271084016 ...

output:

5519324519442957729
13462861144392030499
8898301730697138469
4148979398311169421
15090246851327208067
8628707816538336707
13867297907038176506
10296428352441857103
15654304415409320656
7404566644919251615
9870876264015800597
6356224262148620783
249874952637342736
9023132497407647441
1416175985367538...

result:

ok 200000 lines

Test #35:

score: 19
Accepted
time: 2112ms
memory: 90760kb

input:

200000 200000
11863650499568632095 6646669915142734572 4053375998669045948 14662364203830894482 7319511971537928619 4131498054494361157 7103043576860354080 2730587527012777841 8626012955062792888 7098277606750148625 12990209623538415680 100871355087529915 12267084290544303817 7008468684400973426 856...

output:

11796827338356402281
853302424664654652
564419876379363056
11173501553976506440
2595891684253320024
5590778053561483334
16613470245879883493
5698255402711241072
8125718572356459513
11371252602486101178
7605800585391618993
17939653588292877927
15906636118593883045
14840423959426947118
122409373008752...

result:

ok 200000 lines

Test #36:

score: 19
Accepted
time: 2553ms
memory: 90716kb

input:

200000 200000
7819034666999754227 1261434745131385029 4729051582132640442 9603230030483067185 9188148629137054368 4647584887897271213 14878584596185020248 5036501082632549589 13434605396022727436 10747373329537943689 2859138487129973374 17399126170759425906 5170686633178482234 1548518177806164267 68...

output:

13979872615900358449
13089809959604496472
1561607543806976909
704865086165131395
4750114500765789463
8620101000132483891
1061990993707638791
8219417779659049453
16783172337878697026
731502258552473136
15516870835064396649
13412140913912394465
5029558179142432386
9655856473800271062
14498308915283050...

result:

ok 200000 lines

Test #37:

score: 19
Accepted
time: 2576ms
memory: 90624kb

input:

200000 200000
13667364256171297683 5643454264481609425 13916992424255487273 16139471110620927446 10964822257543206396 18104319795931117637 6754294313610492941 627150985476406559 9787453073837140391 8330282299925244302 16457178623381098023 2644823686047461659 3971198924127091796 9747389824949735337 1...

output:

11367465309833674818
18362877133708215281
7423548724337603968
8957456044849213851
17833913066069915980
7858235162317128668
12543057704351927321
10505054748074388644
6816176482433035300
2467436539272277421
1679282916199502250
4514431222303891247
8020348968469583082
4250522620562980350
344143146282723...

result:

ok 200000 lines

Test #38:

score: 19
Accepted
time: 2184ms
memory: 90884kb

input:

200000 200000
8264785020939323331 5620958879723030311 933008402364228745 5711653520387260744 16167130674961631916 10635243671618608635 7034482071437480120 10254956504177159052 10510387087831623788 8381634740474427698 9506597691312026965 14784451691298216046 15821757099494287606 1919888068508281591 1...

output:

5787014875832445646
9634577796009446262
12314294073540302873
2915216425908688602
7120463906657998875
4286046781319255714
6776880553034928756
7781782119312943753
10261843991161497641
6413565360321098258
15025688638596291097
9526784383328827422
4012177064000612489
1287077077700121461
98702777920648956...

result:

ok 200000 lines

Test #39:

score: 19
Accepted
time: 2471ms
memory: 90568kb

input:

200000 200000
11664667196988092805 1811572170904050995 15694419630875459125 12737549840477675073 16755302998318006416 4014818780481352253 5609118000649893828 6256332194728466258 15733576495075669968 17960532384856428505 15897732465031414620 17425753576410476171 4620624862371502705 112264419736513372...

output:

6480474289914832244
5686011217288555755
14160731859453855234
17554436276709117739
2341367826083254204
8545580044567165676
264741062779916095
3300445074446425091
1133253489203829542
15906618027983611292
4617068730941954223
9183813939934374009
17667722659841632674
17058280131349615456
5995389694319663...

result:

ok 200000 lines

Test #40:

score: 19
Accepted
time: 2357ms
memory: 90628kb

input:

200000 200000
1537940886958943461 9001579450047645548 5527298164056925772 15445874594277229387 11547996012713764596 2685142508745516922 15898551218448062337 16566357055814699599 15778851736432174335 10222916087451023672 14639095824490451029 13130899683058695675 2954207938828868462 150325623383373666...

output:

6794024285749225899
15444253183026756906
16252225983581025191
18270054647591833460
14669804966517502833
7889126920459519979
11162985402720101082
6093427967053788994
10568703419124160247
10842018941076619829
12114458223856608006
16611571295135208112
6660992456255029750
14854289142138250053
8448421034...

result:

ok 200000 lines

Test #41:

score: 19
Accepted
time: 1916ms
memory: 90540kb

input:

200000 200000
9312450956088465409 13091411438344500136 13533801408631028803 397220365455228788 9050318983523848941 14099016315633088068 17718824544128458130 18224580765371017222 16359847342241893914 6209442109030804295 8140737573685484628 16329892452717577856 12435238272437023209 1214903064014288477...

output:

17881136615169673858
10075746974396262535
1101717753396811008
11092187645455964376
312406901260616120
15269190090424605570
16293852523769965660
7287397773622456008
12586683442681630396
10525460771223157010
16483491603350167181
5677129374080663282
17770626863503596216
16241388034502960475
16358076653...

result:

ok 200000 lines

Subtask #7:

score: 0
Skipped

Dependency #1:

0%