結果

問題 No.439 チワワのなる木
ユーザー 🐬hec🐬hec
提出日時 2016-10-29 00:28:58
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,245 bytes
コンパイル時間 1,791 ms
コンパイル使用メモリ 175,392 KB
実行使用メモリ 27,260 KB
最終ジャッジ日時 2024-05-03 08:27:44
合計ジャッジ時間 9,131 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
10,624 KB
testcase_01 AC 4 ms
10,624 KB
testcase_02 AC 4 ms
10,512 KB
testcase_03 AC 4 ms
10,460 KB
testcase_04 AC 3 ms
10,624 KB
testcase_05 AC 4 ms
10,488 KB
testcase_06 AC 3 ms
10,624 KB
testcase_07 AC 4 ms
10,576 KB
testcase_08 AC 3 ms
10,496 KB
testcase_09 AC 5 ms
10,520 KB
testcase_10 AC 5 ms
10,368 KB
testcase_11 AC 4 ms
10,564 KB
testcase_12 AC 5 ms
10,624 KB
testcase_13 AC 4 ms
10,496 KB
testcase_14 AC 5 ms
10,508 KB
testcase_15 AC 5 ms
10,708 KB
testcase_16 AC 6 ms
10,724 KB
testcase_17 AC 5 ms
10,624 KB
testcase_18 AC 87 ms
17,536 KB
testcase_19 AC 94 ms
17,792 KB
testcase_20 AC 122 ms
19,732 KB
testcase_21 AC 37 ms
13,696 KB
testcase_22 AC 29 ms
13,056 KB
testcase_23 AC 131 ms
20,992 KB
testcase_24 AC 134 ms
26,368 KB
testcase_25 TLE -
testcase_26 -- -
testcase_27 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#define _overload(_1,_2,_3,name,...) name
#define _rep(i,n) _range(i,0,n)
#define _range(i,a,b) for(int i=int(a);i<int(b);++i)
#define rep(...) _overload(__VA_ARGS__,_range,_rep,)(__VA_ARGS__)

#define _rrep(i,n) _rrange(i,n,0)
#define _rrange(i,a,b) for(int i=int(a)-1;i>=int(b);--i)
#define rrep(...) _overload(__VA_ARGS__,_rrange,_rrep,)(__VA_ARGS__)

#define _all(arg) begin(arg),end(arg)
#define uniq(arg) sort(_all(arg)),(arg).erase(unique(_all(arg)),end(arg))
#define getidx(ary,key) lower_bound(_all(ary),key)-begin(ary)
#define clr(a,b) memset((a),(b),sizeof(a))
#define bit(n) (1LL<<(n))
#define popcount(n) (__builtin_popcountll(n))

using namespace std;

template<class T>bool chmax(T &a, const T &b) { return (a<b)?(a=b,1):0;}
template<class T>bool chmin(T &a, const T &b) { return (b<a)?(a=b,1):0;}

using ll=long long;
using R=long double;
const R EPS=1e-9; // [-1000,1000]->EPS=1e-8 [-10000,10000]->EPS=1e-7
inline int sgn(const R& r){return(r > EPS)-(r < -EPS);}
inline R sq(R x){return sqrt(max<R>(x,0.0));}

const int dx[8]={1,0,-1,0,1,-1,-1,1};
const int dy[8]={0,1,0,-1,1,1,-1,-1};

// Problem Specific Parameter:

const int nmax=100010;
string s;
vector<int> graph[nmax];
vector<ll> dpc[nmax];
vector<ll> dpw[nmax];


void dfs(int v,int i){
	int p=graph[v][i];
	if(dpc[v][i]>=0) return;
	dpc[v][i]=(s[v]=='c');
	dpw[v][i]=(s[v]=='w');	
	rep(j,graph[v].size()){
		const int v2=graph[v][j];
		if(v2==p) continue;
		int k=lower_bound(begin(graph[v2]),end(graph[v2]),v)-begin(graph[v2]);
		dfs(v2,k);
		dpc[v][i]+=dpc[v2][k];
		dpw[v][i]+=dpw[v2][k];
	}
}

int main(void){
	int n;
	cin >> n >> s;
	
	rep(i,n-1){
		int a,b;
		cin >> a >> b;
		graph[a-1].push_back(b-1);
		graph[b-1].push_back(a-1);
	}

	rep(v,n) sort(begin(graph[v]),end(graph[v]));
	rep(v,n) dpc[v].assign(graph[v].size(),-1),dpw[v].assign(graph[v].size(),-1);
	rep(v,n) rep(i,graph[v].size()) dfs(v,i);

	ll ans=0LL;

	rep(v,n){
		if(s[v]=='c') continue;
		ll tc=0LL,tw=0LL;

		rep(i,graph[v].size()){	
			int v2=graph[v][i];
			int j=lower_bound(begin(graph[v2]),end(graph[v2]),v)-begin(graph[v2]);
			tc+=dpc[v2][j];
			tw+=dpw[v2][j];
			ans-=1LL*dpc[v2][j]*dpw[v2][j];
		}
		ans+=1LL*tc*tw;
	}

	cout << ans << endl;
	return 0;
}
0