QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#825563#9905. 哈夫曼树guleng20070 9ms27440kbC++202.7kb2024-12-21 20:23:222024-12-21 20:23:22

Judging History

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

  • [2024-12-21 20:23:22]
  • 评测
  • 测评结果:0
  • 用时:9ms
  • 内存:27440kb
  • [2024-12-21 20:23:22]
  • 提交

answer

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

const int N=1e5+5;

int son[N*2][2], dist[N*2], fa[N*2], n;
long long a[N*2];

int lc[N*55], rc[N*55], cnt[N*55], root, indx;
long long fir[N*55], lst[N*55];
__int128 sum0[N*55], sum1[N*55], tot;

__int128 Hash(__int128 l,__int128 r)
{
	return l*r;
}

void pushup(int& rt)
{
	if(!lc[rt] && !rc[rt])
	{
		rt=0;
		return;
	}

	if(!lc[rt])
	{
		fir[rt]=fir[rc[rt]], lst[rt]=lst[rc[rt]], sum0[rt]=sum0[rc[rt]], sum1[rt]=sum1[rc[rt]], cnt[rt]=cnt[rc[rt]];
		return;
	}

	if(!rc[rt])
	{
		fir[rt]=fir[lc[rt]], lst[rt]=lst[lc[rt]], sum0[rt]=sum0[lc[rt]], sum1[rt]=sum1[lc[rt]], cnt[rt]=cnt[lc[rt]];
		return;
	}

	fir[rt]=fir[lc[rt]], lst[rt]=lst[rc[rt]], cnt[rt]=cnt[lc[rt]]+cnt[rc[rt]];
	sum0[rt]=sum0[lc[rt]]+(cnt[lc[rt]]%2==0?sum0[rc[rt]]:sum1[rc[rt]]+Hash(lst[lc[rt]],fir[rc[rt]]));
	sum1[rt]=sum1[lc[rt]]+(cnt[lc[rt]]%2==1?sum0[rc[rt]]:sum1[rc[rt]]+Hash(lst[lc[rt]],fir[rc[rt]]));
}

void update(int& rt,long long l,long long r,long long pos,int val)
{
	if(!rt)
		rt=++indx;

	if(l==r)
	{
		cnt[rt] += val;
		if(cnt[rt]==0)
			rt=0;
		else
		{
			fir[rt]=lst[rt]=l;
			sum0[rt]=cnt[rt]/2*(__int128)l*l;
			sum1[rt]=(cnt[rt]-1)/2*(__int128)l*l;
		}
		return;
	}

	long long mid=(l+r)/2;
	if(pos<=mid)
		update(lc[rt],l,mid,pos,val);
	else
		update(rc[rt],mid+1,r,pos,val);
	pushup(rt);
}

void query()
{
	if(sum0[root]==tot)
		printf("YES\n");
	else
		printf("NO\n");
}

void update(int x,long long y)
{
	for(int i=fa[x];i;i=fa[i])
		tot -= Hash(min(a[son[i][0]],a[son[i][1]]),max(a[son[i][0]],a[son[i][1]]));

	if(fa[x])
		update(root,1,1e15,a[x],-1);

	a[x]=y;

	if(fa[x])
		update(root,1,1e15,y,1);

	for(int i=fa[x];i;i=fa[i])
	{
		tot += Hash(min(a[son[i][0]],a[son[i][1]]),max(a[son[i][0]],a[son[i][1]]));
		if(fa[i])
			update(root,1,1e15,a[i],-1);
		a[i]=a[son[i][0]]+a[son[i][1]];
		if(fa[i])
			update(root,1,1e15,a[i],1);
	}
}

int main()
{
	int q;
	cin >> n >> q;
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);

	for(int i=n+1;i<=n*2-1;i++)
	{
		scanf("%d %d",&son[i][0],&son[i][1]);
		fa[son[i][0]]=i, fa[son[i][1]]=i;
	}

	for(int i=n*2-1;i>=n+1;i--)
	{
		dist[son[i][0]]=dist[i]+1;
		dist[son[i][1]]=dist[i]+1;
	}

	for(int i=1;i<=n;i++)
		if(dist[i]>50)
		{
			for(int i=1;i<=q;i++)
				printf("NO\n");
			return 0;
		}

	for(int i=n+1;i<=n*2-1;i++)
	{
		a[i]=a[son[i][0]]+a[son[i][1]];
		update(root,1,1e15,a[son[i][0]],1);
		update(root,1,1e15,a[son[i][1]],1);
		tot += Hash(min(a[son[i][0]],a[son[i][1]]),max(a[son[i][0]],a[son[i][1]]));
	}

	query();
	for(int i=1;i<=q-1;i++)
	{
		int x;
		long long y;
		scanf("%d %lld",&x,&y);
		update(x,y);
		query();
	}
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Subtask #1:

score: 0
Wrong Answer

Test #1:

score: 30
Accepted
time: 0ms
memory: 16040kb

input:

3 4
1 1 1
2 3
1 4
2 2
1 2
3 3

output:

YES
NO
YES
NO

result:

ok 4 token(s): yes count is 2, no count is 2

Test #2:

score: 30
Accepted
time: 2ms
memory: 14140kb

input:

8 5
5 3 4 2 2 6 5 5
1 8
4 5
10 3
11 9
7 2
6 13
14 12
7 3
6 8
4 2
2 5

output:

NO
YES
YES
YES
NO

result:

ok 5 token(s): yes count is 3, no count is 2

Test #3:

score: 30
Accepted
time: 0ms
memory: 22364kb

input:

5 1000
193989534544158 57483670601746 183281373434588 92196008024549 197513473286508
1 5
4 2
7 3
8 6
2 65545142774024
4 67957472319495
5 131478473459166
2 102185858570152
3 191441353035940
5 186000528093501
2 63201184033501
2 77481806092413
3 159789430863849
4 92773786021894
1 194598667478593
3 1458...

output:

YES
YES
YES
NO
NO
NO
NO
NO
NO
YES
YES
YES
YES
YES
YES
YES
YES
NO
NO
YES
YES
YES
YES
YES
YES
NO
NO
NO
NO
YES
NO
YES
YES
YES
YES
YES
YES
NO
NO
NO
NO
YES
YES
YES
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
YES
YES
YES
YES
NO
NO
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
YES
...

result:

ok 1000 token(s): yes count is 375, no count is 625

Test #4:

score: 30
Accepted
time: 9ms
memory: 22988kb

input:

7 1000
88159166205053 95998544558881 48231159865354 231786835189365 84291070100955 225941839972605 33315221625793
2 5
6 4
7 3
1 10
8 11
9 12
6 150843468162951
2 75759088055460
1 86133344610051
4 140694127444493
1 63070113756930
1 90150689680608
6 147790469610032
7 46561924657801
2 103953340734616
6 ...

output:

NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
N...

result:

ok 1000 token(s): yes count is 86, no count is 914

Test #5:

score: 30
Accepted
time: 5ms
memory: 14084kb

input:

7 1000
5 6 7 3 2 5 5
4 7
3 5
1 2
10 6
8 9
12 11
5 2
5 1
5 1
1 2
3 2
5 1
5 2
6 3
4 2
4 1
2 1
5 1
5 1
5 1
7 1
1 2
5 1
5 1
6 3
6 3
5 1
2 2
7 2
7 1
7 1
2 2
1 2
4 2
4 1
1 1
3 1
5 1
2 2
2 2
4 1
2 1
7 2
6 1
2 1
6 2
5 2
1 1
1 1
6 3
7 2
6 3
4 1
1 1
5 1
2 2
7 2
5 1
4 2
5 2
7 2
7 1
4 1
3 2
3 2
1 1
7 2
5 2
1 1
...

output:

NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
YES
YES
YES
YES
NO
YES
NO
YES
YES
YES
YES
YES
YES
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
NO
YES
NO
NO
NO
NO
NO
NO
NO
YES
YES
YES
YES
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO...

result:

ok 1000 token(s): yes count is 199, no count is 801

Test #6:

score: 30
Accepted
time: 0ms
memory: 27440kb

input:

8 1000
172480280419267 212072146993988 23147786306112 161006134777989 37963466387491 50018942108886 18649770050090 50499101532214
7 3
2 1
5 8
9 6
12 11
4 13
10 14
5 57899673021751
7 22087918088240
1 236990257757485
3 21994370580136
7 26427284121647
8 54391900162086
7 27709763713585
6 59631479929866
...

output:

NO
NO
NO
YES
YES
YES
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
YES
NO
NO
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES...

result:

ok 1000 token(s): yes count is 119, no count is 881

Test #7:

score: 30
Accepted
time: 6ms
memory: 24532kb

input:

6 1000
20326746134626 98517002313418 165556087937310 116347939244398 130872245741329 136674587230071
2 3
6 7
8 1
9 5
10 4
4 31902652070998
2 20697837065544
3 14389422108471
5 45879772826060
2 23505088168423
6 7588643182717
2 13873224035289
4 3462171437780
6 12963735380517
6 3348685354818
3 141429739...

output:

NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO...

result:

ok 1000 token(s): yes count is 12, no count is 988

Test #8:

score: 0
Wrong Answer
time: 1ms
memory: 7976kb

input:

72 1000
2 10946 610 987 139583862445 20365011074 5 1597 196418 377 267914296 44945570212853 956722026041 5702887 39088169 75025 27777890035288 12586269025 4181 1 53316291173 144 1548008755920 63245986 1 225851433717 6557470319842 832040 2504730781961 102334155 55 3 46368 7778742049 6765 165580141 72...

output:

NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
...

result:

wrong answer expected YES, found NO [1st token]

Subtask #2:

score: 0
Runtime Error

Test #16:

score: 0
Runtime Error

input:

10000 10000
85117964119 41472951000 61693640396 66409648221 91978532661 62922448518 92497200794 43837348258 45577855926 38256001396 79446271832 95289903258 62510175551 97599809584 56162244722 87617641117 64010325734 56604859803 58544571483 40687963085 38627694508 64665875035 62273927372 73014847094 ...

output:

YES
YES
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
N...

result:


Subtask #3:

score: 0
Runtime Error

Test #20:

score: 0
Runtime Error

input:

50000 50000
16394396247 17456058492 11358090355 13208121167 8612535629 2853817504 18100755237 18603321637 1618810635 7615832530 13631222424 7817630977 10963098997 19654927084 645638016 9352759139 17939720223 15106346489 14475376391 2122412099 15482023637 11224675857 15931143509 4240408932 1270948838...

output:

YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
NO
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
YES
YES
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO...

result:


Subtask #4:

score: 0
Wrong Answer

Test #24:

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

input:

70 100000
66748 126 1 91045172 3605661959574 274077743637349 147314183 8209537 740253 6920630855 25494 1377240316614 15756 6 108000 18118446805 169389361127761 29316262755 48 2643445763 5834083602536 3 9439745562111 29 3719 10 47434709561 11197815949 6018 325122336074 851181326345 1633739329 1527382...

output:

NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
NO
...

result:

wrong answer expected YES, found NO [1st token]