oj.uz

Submission #891557

# Submission time Handle Problem Language Result Execution time Memory
891557 2023-12-23T08:35:54 Z thunopro Sprinkler (JOI22_sprinkler) C++14
3 / 100
 292 ms 77252 KB 
#include<bits/stdc++.h>
using namespace std ; 
#define maxn 300009 
#define ll long long 
#define fi first 
#define se second 
#define pb push_back 
#define left id<<1 
#define right id<<1|1
#define re exit(0);
#define _lower(v,x) lower_bound(v.begin(),v.end(),x)-v.begin()+1

int mod = 1e9+7 ; 
const int INF = 1e9 ;
const int LOG = 18 ; 

typedef vector<int> vi ; 
typedef pair<int,int> pii ; 
typedef vector<pii> vii ;
typedef vector<ll> vl ;  
typedef pair<ll,ll> pll ; 
typedef vector<pll> vll ; 

void add ( int&a , int b ) { if ((a+=b) > mod ) a -= mod ; } 
void sub ( int&a , int b ) { if ((a-=b) < 0 ) a += mod ; } 
template<typename T> void chkmin(T& x, T y) {if(x > y) x = y;}
template<typename T> void chkmax(T& x, T y) {if(x < y) x = y;}

void rf ( ) 
{
	freopen ("bai1.inp","r",stdin) ; 
//	freopen ("bai1.out","w",stdout) ;
}

int _pow ( int a , int n ) 
{
	if ( n == 0 ) return 1 ; 
	int res = _pow ( a , n / 2 ) ; 
	if ( n % 2 ) return ( 1ll * res * res % mod * a % mod ) ; 
	else return ( 1ll * res * res % mod ) ; 
}

int n ;

vi adjList [maxn] ;
int a [maxn] ; 

int nq ; 
struct shape {
	int op , x , d , w ; 
} q [maxn] ; 

bool vis [maxn] ; 
void update_near ( int u , int d , int w , int par = -1 ) 
{
	a [u] = 1ll*a[u]*w%mod ; 
	for ( auto v : adjList [u] ) 
	{
		if ( v != par && d != 0 ) update_near (v,d-1,w,u) ; 
	}
}
void sub1 () 
{
	for ( int rn = 1 ; rn <= nq ; rn ++ ) 
	{
		if ( q[rn].op == 1 ) 
		{
			int x = q [rn].x , d = q [rn].d , w = q [rn].w ; 
			update_near (x,d,w) ; 
		}
		else cout << a [q[rn].x] << "\n" ;  
	}
}

bool check_sub2 () 
{
	for ( int i = 1 ; i <= nq ; i ++ ) 
	{
		if ( q [i].op == 1 && q [i].d >= 2 ) return false ; 
	}
	return true ; 
}


vi vertex_special ; 
int speical [maxn] ;
int lazy_val [maxn] ; 

void sub2 () 
{
	map <int,bool> adj [n+5] ; 
	for ( int i = 1 ; i <= n ; i ++ ) lazy_val [i] = 1 ; 
	for ( int i = 1 ; i <= n ; i ++ ) 
	{
		if ( adjList [i].size () > 0 ) speical [i] = true , vertex_special . pb (i) ; 
		for ( auto x : adjList [i] ) adj [x][i] = adj [i][x] = true ; 
	}
	for ( int rn = 1 ; rn <= nq ; rn ++ ) 
	{
		if ( q[rn].op == 1 ) 
		{
			int x = q [rn].x , d = q [rn].d , w = q [rn].w ; 
			if ( d == 0 ) 
			{
				a [x] = 1ll*a[x]*w%mod ; 
				continue ; 
			}
			if ( speical [x] == false ) update_near (x,d,w) ;
			else lazy_val [x] = 1ll*lazy_val[x]*w%mod ; 	
		}
		else 
		{
			int x = q [rn].x ; 
			int res = 1ll*a [x]*lazy_val[x]%mod ; 
			for ( auto v : vertex_special ) 
			{
				if ( adj [x][v] ) res = 1ll*res*lazy_val[v]%mod ; 
			}
			cout << res << "\n" ; 
		}
	}
}
int main ( ) 
{
	ios_base :: sync_with_stdio (0) ; 
	cin.tie(0) ; cout.tie(0) ;
//	rf () ; 
	cin >> n >> mod ; 
	for ( int i = 1 ; i < n ; i ++ ) 
	{
		int u , v ; cin >> u >> v ; 
		adjList [u] . pb (v) ; 
		adjList [v] . pb (u) ; 
	}
	
	for ( int i = 1 ; i <= n ; i ++ ) cin >> a [i] ; 
	
	cin >> nq ; 
	for ( int i = 1 ; i <= nq ; i ++ ) 
	{
		cin >> q [i].op ; 
		if ( q[i].op == 1 ) 
		{
			cin >> q[i].x >> q[i].d >> q[i].w ; 
		}
		else cin >> q[i].x ; 
	}
	
	if ( n <= 1000 && nq <= 1000 ) sub1 () ; 
	else if ( check_sub2 () ) sub2 () ; 
}




//-std=c++11

Compilation message

sprinkler.cpp: In function 'void rf()':
sprinkler.cpp:31:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   31 |  freopen ("bai1.inp","r",stdin) ;
      |  ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~

Subtask #1
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 11608 KB Output is correct
2 Correct 3 ms 11612 KB Output is correct
3 Correct 2 ms 11612 KB Output is correct
4 Correct 3 ms 11804 KB Output is correct
5 Correct 4 ms 11612 KB Output is correct
6 Correct 6 ms 11612 KB Output is correct
7 Correct 6 ms 11612 KB Output is correct
8 Correct 6 ms 11612 KB Output is correct
9 Correct 3 ms 11612 KB Output is correct
10 Correct 3 ms 11744 KB Output is correct
11 Correct 3 ms 11612 KB Output is correct
12 Correct 2 ms 11708 KB Output is correct
13 Correct 2 ms 11748 KB Output is correct
14 Correct 3 ms 11752 KB Output is correct
15 Correct 2 ms 11612 KB Output is correct
16 Correct 3 ms 11608 KB Output is correct
17 Correct 3 ms 11612 KB Output is correct
18 Correct 3 ms 11612 KB Output is correct
19 Correct 2 ms 11612 KB Output is correct
20 Correct 2 ms 11612 KB Output is correct
21 Correct 2 ms 11612 KB Output is correct
22 Correct 2 ms 11704 KB Output is correct
23 Correct 3 ms 11612 KB Output is correct
24 Correct 3 ms 11616 KB Output is correct
25 Correct 2 ms 11616 KB Output is correct
26 Correct 2 ms 11612 KB Output is correct
27 Correct 3 ms 11616 KB Output is correct
28 Correct 3 ms 11612 KB Output is correct

Subtask #2
0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 11612 KB Output is correct
2 Incorrect 212 ms 58636 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #3
0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 11612 KB Output is correct
2 Incorrect 212 ms 58636 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 11612 KB Output is correct
2 Incorrect 292 ms 77140 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 11620 KB Output is correct
2 Incorrect 281 ms 77252 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 11608 KB Output is correct
2 Correct 3 ms 11612 KB Output is correct
3 Correct 2 ms 11612 KB Output is correct
4 Correct 3 ms 11804 KB Output is correct
5 Correct 4 ms 11612 KB Output is correct
6 Correct 6 ms 11612 KB Output is correct
7 Correct 6 ms 11612 KB Output is correct
8 Correct 6 ms 11612 KB Output is correct
9 Correct 3 ms 11612 KB Output is correct
10 Correct 3 ms 11744 KB Output is correct
11 Correct 3 ms 11612 KB Output is correct
12 Correct 2 ms 11708 KB Output is correct
13 Correct 2 ms 11748 KB Output is correct
14 Correct 3 ms 11752 KB Output is correct
15 Correct 2 ms 11612 KB Output is correct
16 Correct 3 ms 11608 KB Output is correct
17 Correct 3 ms 11612 KB Output is correct
18 Correct 3 ms 11612 KB Output is correct
19 Correct 2 ms 11612 KB Output is correct
20 Correct 2 ms 11612 KB Output is correct
21 Correct 2 ms 11612 KB Output is correct
22 Correct 2 ms 11704 KB Output is correct
23 Correct 3 ms 11612 KB Output is correct
24 Correct 3 ms 11616 KB Output is correct
25 Correct 2 ms 11616 KB Output is correct
26 Correct 2 ms 11612 KB Output is correct
27 Correct 3 ms 11616 KB Output is correct
28 Correct 3 ms 11612 KB Output is correct
29 Correct 2 ms 11612 KB Output is correct
30 Incorrect 212 ms 58636 KB Output isn't correct
31 Halted 0 ms 0 KB -