# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
675302 | vjudge1 | Uzastopni (COCI15_uzastopni) | C++17 | 34 ms | 65536 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
///
#include<bits/stdc++.h>
//#define int long long
#define pii pair<int,int>
#define fi first
#define se second
#pragma GCC tarGet ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
#pragma GCC optimize("Ofast")
#pragma GCC tarGet("avx,avx2,fma")
using namespace std;
using ll = long long;
const int mod = 1e9+7;
const int maxN = 1e4+1;
const int N = 3e5;
const ll infty = 9e18;
void InputFile()
{
//freopen("SHOPPING.inp","r",stdin);
//freopen("SHOPPING.out","w",stdout);
//freopen("test.out","r",stdin);
}
int n;
int a[1001];
vector<int> adj[1001];
bitset<101> dp[1001][101];
vector<pii> valid[1001];
void Read()
{
cin >> n;
for(int i = 1;i <= n;i++)
cin >> a[i];
for(int i = 1;i < n;i++)
{
int x,y;
cin >> x >> y;
adj[x].push_back(y);
adj[y].push_back(x);
}
}
void DFS(int u,int p)
{
for(int v : adj[u])
{
if(v == p) continue;
DFS(v,u);
}
set<pii> st;
for(int v : adj[u])
{
if(v == p) continue;
for(pii x : valid[v])
{
st.insert(x);
}
}
vector<int> save[101];
for(pii tmp : st)
{
save[tmp.fi].push_back(tmp.se);
}
for(int sta = 100;sta >= 1;sta--)
{
if(sta > a[u])
{
for(int p : save[sta])
{
dp[u][sta].set(p);
dp[u][sta] |= dp[u][p+1];
}
}
else if(sta == a[u])
{
dp[u][sta].set(a[u]);
dp[u][sta] |= dp[u][sta+1];
}
else
{
for(int p : save[sta])
{
if(p >= a[u]) break;
dp[u][sta].set(p);
dp[u][sta] |= dp[u][p+1];
}
}
}
for(int sta = 1;sta <= a[u];sta++)
{
for(int j = a[u];j <= 100;j++)
{
if(dp[u][sta][j] == 1) valid[u].emplace_back(sta,j);
}
}
}
void Solve()
{
DFS(1,0);
cout << valid[1].size();
}
int32_t main()
{
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
//InputFile();
//int subtask;
//cin >> subtask;
//Sieve();
//Prepare();
int test;
//cin >> test;
test = 1;
while(test--)
//for(int prc = 1; prc <= test; prc++)
{
Read();
Solve();
//Debug();
}
}
컴파일 시 표준 에러 (stderr) 메시지
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |