QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#718361#9492. 树上简单求和liujiameng0 2162ms106908kbC++144.3kb2024-11-06 20:20:112024-11-06 20:20:14

Judging History

你现在查看的是最新测评结果

  • [2024-11-06 20:20:14]
  • 评测
  • 测评结果:0
  • 用时:2162ms
  • 内存:106908kb
  • [2024-11-06 20:20:11]
  • 提交

answer

#include<bits/stdc++.h>
#define ull unsigned long long
using namespace std;
const int N = 2e5 + 10, D = 18;
vector<int>g1[N],g2[N];
int n;

int sqn, dl[N], dr[N], dfn, l[N], r[N], bel[N], fa[N][D+1], dep1[N];
ull a1[N], a2[N];
void dfs(int x, int f)
{
	dl[x] = ++dfn, fa[x][0] = f, dep1[x] = dep1[f] + 1;
	for(int i=1;i<=D;i++) fa[x][i] = fa[fa[x][i-1]][i-1];
	for(auto j:g1[x]) if(j!=f) dfs(j,x);
	dr[x] = dfn;
}
int LCA(int x, int y)
{
	if(dep1[x]>dep1[y]) swap(x,y);
	int i;
	for(i=D;i>=0;i--) if(dep1[fa[y][i]]>=dep1[x]) y = fa[y][i];
	if(x==y) return x;
	for(i=D;i>=0;i--) if(fa[x][i]!=fa[y][i]) x = fa[x][i], y = fa[y][i];
	return fa[x][0];
}
void init()
{
	dfs(1,0);
	sqn = (int)sqrt(n);
	int i, j;
	for(i=1;r[i-1]!=n;i++) l[i] = r[i-1] + 1, r[i] = min(l[i]+sqn-1,n);
	sqn = i - 1;
	for(i=1;i<=sqn;i++) for(j=l[i];j<=r[i];j++) bel[j] = i;
}
void add(int a, ull x)
{
	a = dl[a];
	int p = bel[a], i;
	for(i=a;i<=r[p];i++) a1[i] += x;
	for(i=p+1;i<=sqn;i++) a2[i] += x;
}
ull ask(int x)
{
	return a1[dr[x]] + a2[bel[dr[x]]] - a1[dl[x]-1] - a2[bel[dl[x]-1]];
}
ull ans[N], val[N];
struct qq{int id,k,x;}qu[N*4],fd[N*4];int cq;
struct op{int x,id;ull y;}ad[N*8];int co;
void ins(int x, int id, ull y){add(x,y),ad[++co]={x,id,y};}

int cnt[N];
int dep[N], p[N], is[N], fa2[N], up[N], faa[N][D+1];
int LCA2(int x, int y)
{
	int i;
	if(dep[x]>dep[y]) swap(x,y);
	for(i=D;i>=0;i--) if(dep[faa[y][i]]>=dep[x]) y = faa[y][i];
	if(x==y) return x;
	for(i=D;i>=0;i--) if(faa[x][i]!=faa[y][i]) x = faa[x][i], y = faa[y][i];
	return faa[x][0];
}
void dfs2(int x, int f)
{
	dep[x] = dep[f] + 1, fa2[x] = faa[x][0] = f;
	for(int i=1;i<=D;i++) faa[x][i] = faa[faa[x][i-1]][i-1];
	for(auto j:g2[x]) if(j!=f) dfs2(j,x);
}
void init2()
{
	dfs2(1,0);
	int i;
	for(i=1;i<=n;i++) p[i] = i;
	sort(&p[1],&p[n]+1,[](int a,int b){return dep[a]>dep[b];});
	for(i=1;i<n;i++)
	{
		int x = p[i], y = fa2[x], z = sqn;
		while(y>1&&z)
		{
			if(is[y]) break;
			y = fa2[y], z--;
		}
		if(!z) is[y] = 1;
	}
	is[1] = 1;
	for(i=2;i<=n;i++) if(!is[i]) 
	{
		int x = fa2[i];
		while(!is[x]) x = fa2[x];
		up[i] = x;
	}
	int c = 0;
	for(i=1;i<=n;i++) c += is[i];
	assert(c<=n/sqn+10);
}

void dfsp(int x, int f)
{
	for(auto j:g2[x]) if(j!=f) cnt[j] += cnt[x], dfsp(j,x);
}

int ww[N];

int main()
{
	int m, i, a, b;
	scanf("%d%d", &n, &m);
	for(i=1;i<=n;i++) scanf("%llu", &val[i]);
	for(i=1;i<n;i++) scanf("%d%d", &a, &b), g1[a].emplace_back(b), g1[b].emplace_back(a);
	for(i=1;i<n;i++) scanf("%d%d", &a, &b), g2[a].emplace_back(b), g2[b].emplace_back(a);
	init();
//	return 0;
	init2();
//	return 0;
	for(i=1;i<=n;i++)
	{
		ins(i,0,val[i]);
		if(i>1) ins(fa[i][0],0,-val[i]);
	}
	for(i=1;i<=m;i++)
	{
		int x, y, xx, yy, l;
		ull k;
		scanf("%d%d%llu", &x, &y, &k), xx = x, yy = y;
		l = LCA(x,y), ins(x,i,k), ins(y,i,k), ins(l,i,-k);
		if(l!=1) ins(fa[l][0],i,-k);
		l = LCA2(x,y);
		while(!is[x]&&x!=l) ans[i] += ask(x), x = fa2[x];
		while(!is[y]&&y!=l) ans[i] += ask(y), y = fa2[y];
		if(!is[x]||!is[y])
		{
			ans[i] += ask(l);
			continue;
		}
		qu[++cq] = {i,1,x}, qu[++cq] = {i,1,y};
		assert(is[x]&&is[y]);
		while(dep[up[x]]>=dep[l]) x = up[x];
		while(dep[up[y]]>=dep[l]) y = up[y];
		qu[++cq] = {i,-1,x}, qu[++cq] = {i,-1,y};
		assert(is[x]&&is[y]);
		if(dep[x]>dep[y]) swap(x,y);
		int jump = 0;
		while(dep[x]<dep[y]) ans[i] += ask(y), y = fa2[y], jump++;
		while(x!=y) ans[i] += ask(x) + ask(y), x = fa2[x], y = fa2[y], jump++;
//		assert(jump<=sqn*2+10);
		ans[i] += ask(x);
	}
	for(i=1;i<=cq;i++) ww[qu[i].x]++;
	for(i=1;i<=n;i++) ww[i] += ww[i-1];
	for(i=cq;i>=1;i--) fd[ww[qu[i].x]--] = qu[i];
	for(i=1;i<=cq;i++) qu[i] = fd[i];
	for(i=1;i<cq;i++) assert(qu[i].x<qu[i+1].x||(qu[i].x==qu[i+1].x&&qu[i].id<=qu[i+1].id));
	int aaa = 0, l, r;
	for(i=1;i<=cq;i=r+1)
	{
		l = i, r = i;
		int x, j;
		ull res = 0;
		while(r<=cq&&qu[r].x==qu[l].x) r++;
		r--, x = qu[l].x;
//		if(!is[x]) cout<<"sfdahsfda\n";
		aaa++;
//		continue;
		for(i=1;i<=n;i++) cnt[i] = 0;
		for(i=x;i;i=fa2[i]) cnt[i] = 1;
		dfsp(1,0);
		continue;
		for(i=l,j=1;i<=r;i++)
		{
			while(j<=co&&ad[j].id<=qu[i].id) res += ad[j].y * cnt[ad[j].x], j++;
			ans[qu[i].id] += res * qu[j].k;
		}
	}
	assert(aaa<=n/sqn+10);
	for(i=1;i<=m;i++) printf("%llu\n", ans[i]);
	return 0;
}

详细

Subtask #1:

score: 0
Wrong Answer

Test #1:

score: 0
Wrong Answer
time: 3ms
memory: 30816kb

input:

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

output:

12105153858659381124
14289412666719262474
11668241962484097878
11288238120352358472
16638191024235179689
5864805956833289388
18083038299337574635
1327700673955047890
14984128302052618266
7991618764703973546
1938481927132632884
12179778017143396902
8569443169249732820
14475184194298579044
48941469887...

result:

wrong answer 2nd lines differ - expected: '18367442707572066757', found: '14289412666719262474'

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: 1744ms
memory: 99584kb

input:

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

output:

9042998055336671259
14937416548619115396
5835924579879681322
9187084356907537870
17810346410706951073
565636160725988981
12105441902314467382
16059573289829807099
7246210357888652619
7725251776483176497
17088098442183693937
9042305714006927228
5989073646081200700
6526772063609981609
5157820245646960...

result:

wrong answer 2nd lines differ - expected: '11611293489264521142', found: '14937416548619115396'

Subtask #5:

score: 0
Time Limit Exceeded

Test #27:

score: 0
Time Limit Exceeded

input:

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

output:


result:


Subtask #6:

score: 0
Wrong Answer

Test #34:

score: 0
Wrong Answer
time: 2162ms
memory: 106908kb

input:

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

output:

16678427573746718672
17865923446737938161
8898301730697138469
7164088435544405543
10716760066985972442
7997477085493320374
3220363492177892799
902815997751267838
2261630925866102477
12072531807864605185
3730763216504399702
18020034424712210734
12199719899175687164
13933090542618095403
12460289540427...

result:

wrong answer 1st lines differ - expected: '5519324519442957729', found: '16678427573746718672'

Subtask #7:

score: 0
Skipped

Dependency #1:

0%