結果

問題 No.439 チワワのなる木
ユーザー char134217728char134217728
提出日時 2017-09-15 03:48:04
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 53 ms / 5,000 ms
コード長 2,181 bytes
コンパイル時間 2,356 ms
コンパイル使用メモリ 148,444 KB
実行使用メモリ 21,592 KB
最終ジャッジ日時 2023-08-07 17:17:35
合計ジャッジ時間 3,423 ms
ジャッジサーバーID
(参考情報)
judge13 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,836 KB
testcase_01 AC 3 ms
5,856 KB
testcase_02 AC 3 ms
5,728 KB
testcase_03 AC 3 ms
5,672 KB
testcase_04 AC 3 ms
5,736 KB
testcase_05 AC 3 ms
5,748 KB
testcase_06 AC 3 ms
5,956 KB
testcase_07 AC 3 ms
5,680 KB
testcase_08 AC 4 ms
5,660 KB
testcase_09 AC 3 ms
5,672 KB
testcase_10 AC 3 ms
5,700 KB
testcase_11 AC 3 ms
5,864 KB
testcase_12 AC 4 ms
5,648 KB
testcase_13 AC 3 ms
5,684 KB
testcase_14 AC 3 ms
5,764 KB
testcase_15 AC 4 ms
5,716 KB
testcase_16 AC 4 ms
5,800 KB
testcase_17 AC 4 ms
5,784 KB
testcase_18 AC 34 ms
8,024 KB
testcase_19 AC 30 ms
8,040 KB
testcase_20 AC 43 ms
8,732 KB
testcase_21 AC 13 ms
6,988 KB
testcase_22 AC 13 ms
6,516 KB
testcase_23 AC 47 ms
10,048 KB
testcase_24 AC 53 ms
19,300 KB
testcase_25 AC 34 ms
9,384 KB
testcase_26 AC 32 ms
9,200 KB
testcase_27 AC 40 ms
21,592 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:81:6: warning: ISO C++ forbids declaration of ‘main’ with no type [-Wreturn-type]
 main(){
      ^

ソースコード

diff #

#include <bits/stdc++.h>
#define FOR(i,a,b) for (int i=(a);i<(b);i++)
#define FORR(i,a,b) for (int i=(a);i>=(b);i--)
#define pb push_back
#define pcnt __builtin_popcount
#define show(x) cout<<#x<<" = "<<x<<endl;
#define maxs(x,y) x = max(x,y)
#define mins(x,y) x = min(x,y)
#define fi first
#define se second
#define rng(a) a.begin(),a.end()
#define each(it,c) for(__typeof((c).begin()) it=(c).begin();it!=(c).end();it++)
#define sz(x) (int)(x).size()
#define mp make_pair

using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<pii> vpii;
typedef set<int> si;
typedef pair<ll,ll> pll;
typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef vector<pll> vpll;
typedef set<ll> sl;
template<typename T>string join(vector<T>&v)
{stringstream s;FOR(i,0,sz(v))s<<' '<<v[i];return s.str().substr(1);}
ll gcd(ll a,ll b){if(a>b)swap(a,b);for(;a>0;b%=a,swap(a,b));return b;}
int modpow(ll a,ll n,int m){if(a==0)return a;ll p=1;for(;n>0;n/=2,a=a*a%m)if(n&1)p=p*a%m;return(int)p;}
void dout(double d){printf("%.12f\n",d);}

const int iinf = 1e9;
const ll linf = 1e18;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-10;
int n;
string s;
ll ans;
vi E[100005];

typedef pair<pll, pll> ql;
ql search(int a, int p){
  ll c = 0, w = 0, cw = 0, ww = 0;
  ll sc = 0, sw = 0, scw = 0, sww = 0;
  each(itr, E[a]){
    if(*itr == p)continue;
    ql r = search(*itr, a);
    ans += r.fi.fi * sww;
    ans += r.fi.se * scw;
    ans += r.se.fi * sw;
    ans += r.se.se * sc;
    if(s[a] == 'c'){
      ans += r.se.se;
    }else{
      ans += r.se.fi;
      ans += sc * r.fi.se;
      ans += sw * r.fi.fi;
    }
    sc += r.fi.fi;
    sw += r.fi.se;
    scw += r.se.fi;
    sww += r.se.se;
  }
  if(s[a] == 'c'){
    c = sc + 1;
    w = sw;
    cw = scw;
    ww = sww;
  }else{
    c = sc;
    w = sw + 1;
    cw = scw + sc;
    ww = sww + sw;
  }
  return mp(mp(c,w),mp(cw,ww));
}

main(){
  cin.tie(0);
  ios::sync_with_stdio(false);
  cin >> n >> s;
  FOR(i, 1, n){
    int a, b;
    cin >> a >> b;
    a--;b--;
    E[a].pb(b);
    E[b].pb(a);
  }
  search(0, -1);
  cout << ans << endl;
  return 0;
}
0