結果

問題 No.1124 Earthquake Safety
ユーザー cn_449cn_449
提出日時 2020-07-24 19:42:35
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,092 bytes
コンパイル時間 1,131 ms
コンパイル使用メモリ 119,412 KB
実行使用メモリ 47,680 KB
最終ジャッジ日時 2024-06-25 18:09:01
合計ジャッジ時間 14,758 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 58
権限があれば一括ダウンロードができます

ソースコード

diff # 

//後で解く
#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <iomanip>
#include <utility>
#include <tuple>
#include <functional>
#include <bitset>
#include <cassert>
#include <complex>
#include <stdio.h>
#include <time.h>
#include <numeric>
#include <random>
#include <unordered_map>
#include <unordered_set>
#define all(a) a.begin(),a.end()
#define rep(i, n) for (ll i = 0; i < (n); i++)
#define pb push_back
#define debug(x) cerr << #x << ':' << x << '\n'
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<ll, ll> P;
typedef complex<ld> com;
constexpr int inf = 1000000010;
constexpr ll INF = 1000000000000000010;
constexpr ld eps = 1e-12;
constexpr ld pi = 3.141592653589793238;
template<class T, class U> inline bool chmax(T &a, const U &b) { if (a < b) { a = b; return true; } return false; }
template<class T, class U> inline bool chmin(T &a, const U &b) { if (a > b) { a = b; return true; } return false; }

constexpr ll mod = 1000000007;

ll half(ll n) {
	if (n & 1) n += mod;
	n >>= 1;
	return n;
}

void add(ll &a, ll b) {
	a += b;
	if (a >= mod) a -= mod;
	if (a < 0) a += mod;
}

constexpr int sz = 300010;
vector<vector<int>> graph(sz, vector<int>());
vector<ll> memox(sz, 1), memoy(sz, 1);

void dfs1(int n, int p) {
	for (int i : graph[n]) {
		if (i == p) continue;
		dfs1(i, n);
		add(memox[n], half(memox[i]));
	}
}

void dfs2(int n, int p) {
	ll sum = 0;
	for (int i : graph[n]) {
		if (i == p) continue;
		add(memoy[i], half(memoy[n]));
		sum += half(memox[i]);
	}
	sum %= mod;
	for (int i : graph[n]) {
		if (i == p) continue;
		add(memoy[i], sum - half(memox[i]));
		dfs2(i, n);
	}
}


int main() {
	cin.tie(0);
	ios::sync_with_stdio(false);
	cout << fixed << setprecision(20);

	int n;
	cin >> n;
	rep(i, n - 1) {
		int u, v;
		cin >> u >> v;
		u--; v--;
		graph[u].pb(v);
		graph[v].pb(u);
	}
	dfs1(0, -1);
	dfs2(0, -1);
}
0