結果

問題 No.439 チワワのなる木
コンテスト
ユーザー 🐬hec
提出日時 2016-10-29 14:16:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 116 ms / 5,000 ms
コード長 1,823 bytes
コンパイル時間 1,763 ms
コンパイル使用メモリ 172,188 KB
実行使用メモリ 20,044 KB
最終ジャッジ日時 2024-11-24 17:51:38
合計ジャッジ時間 3,262 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

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];
ll ans=0LL,tc=0LL,tw=0LL;

tuple<ll,ll> dfs(int v,int p){
	ll cc=0LL,cw=0LL;
	if(s[v]=='w') ans+=1LL*tc*(tw-1);

	for(auto &v2:graph[v]){
		if(v2==p) continue;
		ll ac,aw;
		tie(ac,aw)=dfs(v2,v);
		if(s[v]=='w') ans-=1LL*ac*aw;
		cc+=ac,cw+=aw;
	}

	if(s[v]=='w' and p!=-1) ans-=1LL*(tc-cc)*(tw-1-cw);
	if(s[v]=='c') cc++;
	if(s[v]=='w') cw++;

	return tie(cc,cw);
}

int main(void){
	int n;
	cin >> n >> s;
	rep(i,n) tc+=(s[i]=='c'),tw+=(s[i]=='w');


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

	dfs(0,-1);
	cout << ans << endl;
	return 0;
}
0