oj.uz

답안 #778629

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
778629 2023-07-10T13:48:07 Z DangerNoodle7591 Deblo (COCI18_deblo) C++17
18 / 90
 86 ms 56736 KB 
#include<bits/stdc++.h>
using namespace std;
#define lalala ios_base::sync_with_stdio(false);cin.tie(NULL);
#define endl "\n"
#define ll long long
#define pb push_back
#define N 100005
#define M 26
vector<int> adj[N];
ll int dp[N][M][2],arr[N];
ll int cev=0;
void dfs(int x,int once){
	for(int i=0;i<M;i++){
		if((arr[x]>>i)&1)dp[x][i][1]=1;
		else dp[x][i][0]=1;
	}
	for(auto u:adj[x]){
		if(u==once)continue;
		dfs(u,x);
		for(int i=0;i<M;i++){
			ll int a=dp[u][i][0],b=dp[u][i][1];
			int top=(dp[x][i][0]*b+dp[x][i][1]*a);
			cev+=(top<<i);
			int k=(arr[x]>>i)&1;
			dp[x][i][k]+=a;
			dp[x][i][k^1]+=b;
		}
	}
}
int main(){
	lalala;
	int n;cin>>n;
	for(int i=1;i<=n;i++){
		cin>>arr[i];
		cev+=arr[i];
	}
	for(int i=0;i<n-1;i++){
		int a,b;cin>>a>>b;
		adj[a].pb(b);adj[b].pb(a);
	}
	dfs(1,-1);
	cout<<cev<<endl;
 
}

Subtask #1
18.0 / 90.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 2644 KB Output is correct
2 Correct 2 ms 2676 KB Output is correct
3 Incorrect 2 ms 2644 KB Output isn't correct
4 Incorrect 3 ms 3072 KB Output isn't correct
5 Incorrect 2 ms 3028 KB Output isn't correct
6 Incorrect 61 ms 56736 KB Output isn't correct
7 Incorrect 64 ms 56672 KB Output isn't correct
8 Incorrect 66 ms 49172 KB Output isn't correct
9 Incorrect 68 ms 48300 KB Output isn't correct
10 Incorrect 86 ms 47496 KB Output isn't correct