結果
| 問題 | No.235 めぐるはめぐる (5) |
| ユーザー |
 anta
|
| 提出日時 | 2015-06-27 14:55:37 |
| 言語 | C++11 (gcc 11.4.0) |
| 結果 |
AC
|
| 実行時間 | 1,003 ms / 10,000 ms |
| コード長 | 10,849 bytes |
| コンパイル時間 | 1,461 ms |
| コンパイル使用メモリ | 112,716 KB |
| 実行使用メモリ | 19,072 KB |
| 最終ジャッジ日時 | 2024-07-07 19:59:03 |
| 合計ジャッジ時間 | 5,549 ms |
|
ジャッジサーバーID （参考情報） |
judge4 / judge2 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 1,003 ms
19,072 KB |
| testcase_01 | AC | 407 ms
18,944 KB |
| testcase_02 | AC | 842 ms
18,816 KB |
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:409:32: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
409 | rep(i, N) scanf("%d", &S[i]);
| ~~~~~^~~~~~~~~~~~~
main.cpp:410:32: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
410 | rep(i, N) scanf("%d", &C[i]);
| ~~~~~^~~~~~~~~~~~~
main.cpp:421:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
421 | scanf("%d%d", &A, &B), -- A, -- B;
| ~~~~~^~~~~~~~~~~~~~~~
main.cpp:425:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
425 | scanf("%d", &Q);
| ~~~~~^~~~~~~~~~
main.cpp:428:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
428 | scanf("%d", &ty);
| ~~~~~^~~~~~~~~~~
main.cpp:431:38: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
431 | scanf("%d%d%d", &X, &Y, &Z), -- X, -- Y;
| ~~~~~^~~~~~~~~~~~~~~~~~~~~~
main.cpp:437:38: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
437 | scanf("%d%d", &X, &Y), -- X, -- Y;
| ~~~~~^~~~~~~~~~~~~~~~
ソースコード
#include <string>
#include <vector>
#include <algorithm>
#include <numeric>
#include <set>
#include <map>
#include <queue>
#include <iostream>
#include <sstream>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstring>
#include <cctype>
#include <cassert>
#include <limits>
#include <functional>
#define rep(i,n) for(int (i)=0;(i)<(int)(n);++(i))
#define rer(i,l,u) for(int (i)=(int)(l);(i)<=(int)(u);++(i))
#define reu(i,l,u) for(int (i)=(int)(l);(i)<(int)(u);++(i))
#if defined(_MSC_VER) || __cplusplus > 199711L
#define aut(r,v) auto r = (v)
#else
#define aut(r,v) __typeof(v) r = (v)
#endif
#define each(it,o) for(aut(it, (o).begin()); it != (o).end(); ++ it)
#define all(o) (o).begin(), (o).end()
#define pb(x) push_back(x)
#define mp(x,y) make_pair((x),(y))
#define mset(m,v) memset(m,v,sizeof(m))
#define INF 0x3f3f3f3f
#define INFL 0x3f3f3f3f3f3f3f3fLL
using namespace std;
typedef vector<int> vi; typedef pair<int,int> pii; typedef vector<pair<int,int> > vpii; typedef long long ll;
template<typename T, typename U> inline void amin(T &x, U y) { if(y < x) x = y; }
template<typename T, typename U> inline void amax(T &x, U y) { if(x < y) x = y; }
struct CentroidPathDecomposition {
vector<int> colors, positions; //Vertex -> Color, Vertex -> Offset
vector<int> lengths, parents, branches; //Color -> Int, Color -> Color, Color -> Offset
vector<int> parentnodes, depths; //Vertex -> Vertex, Vertex -> Int
//vector<FenwickTree>とかを避けて1次元にしたい時に使う
//sortednodesの[lefts[v], rights[v])はvのsubtreeとなっている
vector<int> sortednodes, offsets; //Index -> Vertex, Color -> Index
vector<int> lefts, rights; //Vertex -> Index
struct BuildDFSState {
int i, len, parent;
BuildDFSState() { }
BuildDFSState(int i_, int l, int p): i(i_), len(l), parent(p) { }
};
//両方の辺があってもいいし、親から子への辺だけでもよい
void build(const vector<vi> &g, int root) {
int n = g.size();
colors.assign(n, -1); positions.assign(n, -1);
lengths.clear(); parents.clear(); branches.clear();
parentnodes.assign(n, -1); depths.assign(n, -1);
sortednodes.clear(); offsets.clear();
lefts.assign(n, -1); rights.assign(n, -1);
vector<int> subtreesizes;
measure(g, root, subtreesizes);
typedef BuildDFSState State;
depths[root] = 0;
vector<State> s;
s.push_back(State(root, 0, -1));
while(!s.empty()) {
State t = s.back(); s.pop_back();
int i = t.i, len = t.len;
int index = sortednodes.size();
int color = lengths.size();
if(t.parent == -3) {
rights[i] = index;
continue;
}
if(t.parent != -2) {
assert(parents.size() == color);
parents.push_back(t.parent);
branches.push_back(len);
offsets.push_back(index);
len = 0;
}
colors[i] = color;
positions[i] = len;
lefts[i] = index;
sortednodes.push_back(i);
int maxsize = -1, maxj = -1;
each(j, g[i]) if(colors[*j] == -1) {
if(maxsize < subtreesizes[*j]) {
maxsize = subtreesizes[*j];
maxj = *j;
}
parentnodes[*j] = i;
depths[*j] = depths[i] + 1;
}
s.push_back(State(i, -1, -3));
if(maxj == -1) {
lengths.push_back(len + 1);
}else {
each(j, g[i]) if(colors[*j] == -1 && *j != maxj)
s.push_back(State(*j, len, color));
s.push_back(State(maxj, len + 1, -2));
}
}
}
void get(int v, int &c, int &p) const {
c = colors[v]; p = positions[v];
}
bool go_up(int &c, int &p) const {
p = branches[c]; c = parents[c];
return c != -1;
}
inline const int *nodesBegin(int c) const { return &sortednodes[0] + offsets[c]; }
inline const int *nodesEnd(int c) const { return &sortednodes[0] + (c+1 == offsets.size() ? sortednodes.size() : offsets[c+1]); }
private:
void measure(const vector<vi> &g, int root, vector<int> &out_subtreesizes) const {
out_subtreesizes.assign(g.size(), -1);
vector<int> s;
s.push_back(root);
while(!s.empty()) {
int i = s.back(); s.pop_back();
if(out_subtreesizes[i] == -2) {
int s = 1;
each(j, g[i]) if(out_subtreesizes[*j] != -2)
s += out_subtreesizes[*j];
out_subtreesizes[i] = s;
}else {
s.push_back(i);
each(j, g[i]) if(out_subtreesizes[*j] == -1)
s.push_back(*j);
out_subtreesizes[i] = -2;
}
}
}
};
template<int MOD>
struct ModInt {
static const int Mod = MOD;
unsigned x;
ModInt(): x(0) { }
ModInt(signed sig) { int sigt = sig % MOD; if(sigt < 0) sigt += MOD; x = sigt; }
ModInt(signed long long sig) { int sigt = sig % MOD; if(sigt < 0) sigt += MOD; x = sigt; }
int get() const { return (int)x; }
ModInt &operator+=(ModInt that) { if((x += that.x) >= MOD) x -= MOD; return *this; }
ModInt &operator-=(ModInt that) { if((x += MOD - that.x) >= MOD) x -= MOD; return *this; }
ModInt &operator*=(ModInt that) { x = (unsigned long long)x * that.x % MOD; return *this; }
ModInt &operator/=(ModInt that) { return *this *= that.inverse(); }
ModInt operator+(ModInt that) const { return ModInt(*this) += that; }
ModInt operator-(ModInt that) const { return ModInt(*this) -= that; }
ModInt operator*(ModInt that) const { return ModInt(*this) *= that; }
ModInt operator/(ModInt that) const { return ModInt(*this) /= that; }
ModInt inverse() const {
signed a = x, b = MOD, u = 1, v = 0;
while(b) {
signed t = a / b;
a -= t * b; std::swap(a, b);
u -= t * v; std::swap(u, v);
}
if(u < 0) u += Mod;
ModInt res; res.x = (unsigned)u;
return res;
}
bool operator==(ModInt that) const { return x == that.x; }
bool operator!=(ModInt that) const { return x != that.x; }
ModInt operator-() const { ModInt t; t.x = x == 0 ? 0 : Mod - x; return t; }
};
typedef ModInt<1000000007> mint;
vector<mint> coefs, coefsum;
struct Val {
mint val;
mint coef;
Val() { }
};
struct Sum {
mint sum, coefsum;
Sum() { }
Sum(const Val &val): sum(val.val), coefsum(val.coef) { }
Sum &operator+=(const Sum &that) { sum += that.sum; coefsum += that.coefsum; return *this; }
Sum operator+(const Sum &that) const { return Sum(*this) += that; }
};
struct Add {
mint add;
Add() { }
Add &operator+=(const Add &that) { add += that.add; return *this; }
void addToVal(Val &val) const { val.val += add * val.coef; }
void addToSum(Sum &sum) const { sum.sum += add * sum.coefsum; }
operator bool() const { return add.x != 0; }
};
struct Node {
private:
Node *parent, *left, *right;
Val val; Sum sum; Add add;
bool rev;
public:
Node(): parent(NULL), left(NULL), right(NULL),
val(), sum(Val()), add(), rev(false) { }
private:
bool is_root() const {
return !parent || (parent->left != this && parent->right != this);
}
void update() {
sum = Sum(val);
if(left) {
left->propagate();
sum += left->sum;
}
if(right) {
right->propagate();
sum += right->sum;
}
}
void propagate() {
if(rev) {
if(left) left->rev = !left->rev;
if(right) right->rev = !right->rev;
swap(left, right);
rev = false;
}
if(add) {
add.addToVal(val);
add.addToSum(sum);
if(left) left->add += add;
if(right) right->add += add;
add = Add();
}
}
public:
bool debugCheckUpdated() const {
Node tmp = *this;
tmp.update();
return memcmp(this, &tmp, sizeof(Node)) == 0;
}
bool debugCheckPropagated() const {
return !rev && !add;
}
private:
void rotateR() {
propagate();
Node *q = parent, *r = q->parent;
if(q->left = right) right->parent = q;
right = q; q->parent = this;
if(parent = r) {
if(r->left == q) r->left = this;
if(r->right == q) r->right = this;
}
q->update(); update();
}
void rotateL() {
propagate();
Node *q = parent, *r = q->parent;
if(q->right = left) left->parent = q;
left = q; q->parent = this;
if(parent = r) {
if(r->left == q) r->left = this;
if(r->right == q) r->right = this;
}
q->update(); update();
}
void topdown() {
static vector<Node*> route;
for(Node *q = this; ; q = q->parent) {
route.push_back(q);
if(q->is_root()) break;
}
for(int i = route.size()-1; i >= 0; i --) {
Node *q = route[i];
q->propagate();
}
route.clear();
}
void splay() {
topdown();
while(!is_root()) {
Node *q = parent;
if(q->is_root()) {
if(q->left == this) rotateR();
else rotateL();
}else {
Node *r = q->parent;
if(r->left == q) {
if(q->left == this) q->rotateR(), rotateR();
else rotateL(), rotateR();
}else {
if(q->right == this) q->rotateL(), rotateL();
else rotateR(), rotateL();
}
}
}
}
public:
void setVal(const Val &val_) {
assert(is_root());
val = val_;
update();
}
void addAdd(const Add &add_) {
assert(is_root());
add += add_;
}
Val getVal() const {
assert(is_root());
Val t = val;
add.addToVal(t);
return t;
}
Sum getSum() const {
assert(is_root() && debugCheckUpdated());
Sum t = sum;
add.addToSum(t);
return t;
}
private:
static Node *pathHead(Node *a) {
assert(a->is_root() && a->debugCheckPropagated());
Node *h = a;
while(1) {
Node *c = h->left;
if(!c) break;
c->propagate();
h = c;
}
h->splay();
return h;
}
static void splitPath(Node *a) {
assert(a->is_root() && a->debugCheckPropagated());
Node *r = a->right;
if(r != NULL) {
a->right = NULL;
a->update();
}
}
public:
static void expose(Node *x) {
Node *rp = NULL;
for(Node *p = x; p != NULL; p = p->parent) {
p->splay();
p->right = rp;
p->update();
rp = p;
}
x->splay();
}
static void exposePath(Node *a, Node *b) {
evert(a);
a->propagate();
expose(b);
assert(a == b || a->parent != NULL);
splitPath(b);
}
static void evert(Node *x) {
expose(x);
splitPath(x);
assert(x->debugCheckPropagated());
x->rev = true;
}
static void connect(Node *x, Node *y) {
evert(x);
x->parent = y;
}
};
bool naivegetpath(int i, int p, int t, const vector<vi> &g, vi &path) {
bool r = false;
if(i == t) {
r = true;
}else {
each(j, g[i]) if(*j != p)
r = r || naivegetpath(*j, i, t, g, path);
}
if(r)
path.push_back(i);
return r;
}
int main() {
int N;
while(1) {
if(!~scanf("%d", &N)) break;
vector<int> S(N), C(N);
rep(i, N) scanf("%d", &S[i]);
rep(i, N) scanf("%d", &C[i]);
vector<Node> nodes(N);
rep(i, N) {
Val val;
val.val = S[i];
val.coef = C[i];
nodes[i].setVal(val);
}
vector<vi> g(N);
rep(i, N-1) {
int A, B;
scanf("%d%d", &A, &B), -- A, -- B;
Node::connect(&nodes[A], &nodes[B]);
}
int Q;
scanf("%d", &Q);
rep(ii, Q) {
int ty;
scanf("%d", &ty);
if(ty == 0) {
int X, Y, Z;
scanf("%d%d%d", &X, &Y, &Z), -- X, -- Y;
Add add; add.add = Z;
Node::exposePath(&nodes[X], &nodes[Y]);
nodes[Y].addAdd(add);
}else {
int X, Y;
scanf("%d%d", &X, &Y), -- X, -- Y;
Node::exposePath(&nodes[X], &nodes[Y]);
mint ans = nodes[Y].getSum().sum;
printf("%d\n", ans.get());
}
}
}
return 0;
}