#389326 (C++14) No.898 tri-βutree

• 問題: No.898 tri-βutree
• ユーザー: kotatsugame
• 提出日時: 2019-10-12 02:22:26
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 489 ms / 4,000 ms
• コード長: 1,132 bytes
• コンパイル時間: 1,007 ms
• コンパイル使用メモリ: 71,680 KB
• 実行使用メモリ: 23,936 KB
• 最終ジャッジ日時: 2024-11-08 23:06:11
• 合計ジャッジ時間: 10,856 ms
• ジャッジサーバーID （参考情報）: judge2 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	298 ms; 23,936 KB
testcase_01	AC	5 ms; 6,656 KB
testcase_02	AC	5 ms; 6,656 KB
testcase_03	AC	5 ms; 6,784 KB
testcase_04	AC	5 ms; 6,656 KB
testcase_05	AC	5 ms; 6,656 KB
testcase_06	AC	6 ms; 6,656 KB
testcase_07	AC	477 ms; 18,304 KB
testcase_08	AC	489 ms; 18,176 KB
testcase_09	AC	489 ms; 18,048 KB
testcase_10	AC	482 ms; 18,176 KB
testcase_11	AC	471 ms; 18,176 KB
testcase_12	AC	483 ms; 18,176 KB
testcase_13	AC	481 ms; 18,304 KB
testcase_14	AC	473 ms; 18,048 KB
testcase_15	AC	487 ms; 18,304 KB
testcase_16	AC	479 ms; 18,304 KB
testcase_17	AC	484 ms; 18,048 KB
testcase_18	AC	472 ms; 18,176 KB
testcase_19	AC	477 ms; 18,176 KB
testcase_20	AC	479 ms; 18,176 KB
testcase_21	AC	486 ms; 18,176 KB

コンパイルメッセージ

main.cpp:38:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   38 | main()
      | ^~~~

ソースコード

#include<iostream>
#include<vector>
using namespace std;
int N;
vector<pair<int,int> >G[1<<17];
int depth[1<<17];
long weighth[1<<17];
int parent[17][1<<17];
void dfs(int u,int p,int d,long wei)
{
	depth[u]=d;
	weighth[u]=wei;
	parent[0][u]=p;
	for(pair<int,int>q:G[u])
	{
		if(q.first==p)continue;
		dfs(q.first,u,d+1,wei+q.second);
	}
}
int lca(int u,int v)
{
	if(depth[u]>depth[v])swap(u,v);
	for(int k=16;k>=0;k--)
	{
		if(depth[v]-depth[u]>>k&1)v=parent[k][v];
	}
	if(u==v)return u;
	for(int k=16;k>=0;k--)
	{
		if(parent[k][u]!=parent[k][v])
		{
			u=parent[k][u];
			v=parent[k][v];
		}
	}
	return parent[0][u];
}
main()
{
	cin>>N;
	for(int i=1;i<N;i++)
	{
		int u,v,w;cin>>u>>v>>w;
		G[u].push_back(make_pair(v,w));
		G[v].push_back(make_pair(u,w));
	}
	dfs(0,-1,0,0);
	for(int k=1;k<17;k++)
	{
		for(int i=0;i<N;i++)
		{
			if(parent[k-1][i]>=0)parent[k][i]=parent[k-1][parent[k-1][i]];
			else parent[k][i]=-1;
		}
	}
	int Q;
	cin>>Q;
	for(;Q--;)
	{
		int x,y,z;cin>>x>>y>>z;
		int xy=lca(x,y);
		int yz=lca(y,z);
		int zx=lca(z,x);
		cout<<weighth[x]+weighth[y]+weighth[z]-weighth[xy]-weighth[yz]-weighth[zx]<<endl;
	}
}