#include <vector>
#include<bits/stdc++.h>
#define f first
#define s second
using namespace std;
const int mx=2e5+12;
vector<pair<int,long long>> g[mx];
long long d[mx];
long long d1[mx];
long long d2[mx];
int p1[mx];
int p2[mx];
int A[mx];
long long n,k;
bool ok=0;
void dfs1(int v,int p,int d,int mask){
d=min(d,A[v]);
if(!d && A[v]){
ok=1;
}
for(auto To:g[v]){
if(To.f!=p){
dfs1(To.f,v,d,mask);
}
}
}
void dfs(int v,int p,bool ok){
for(auto To:g[v]){
int to=To.f,w=To.s;
if(to==p){
continue;
}
if(ok){
d2[to]=d2[v]+w;
}
else{
d1[to]=d1[v]+w;
}
dfs(to,v,ok);
}
}
int max_score(int N, int x, int y, long long K, std::vector<int> U,std::vector<int> V, std::vector<int> W){
n=N,k=K;
for(int i=0;i<n;i++){
g[i].clear();
d1[i]=d2[i]=0;
}
bool ok=1;
for(int i=0;i<n-1;i++){
if(!(V[i]==i && U[i]==i+1)){
ok=0;
}
g[V[i]].push_back({U[i],W[i]});
g[U[i]].push_back({V[i],W[i]});
}
if(ok){
dfs(x,-1,0);
dfs(y,-1,1);
long long ans=0;
for(int l1=x;l1>=0;l1--){
for(int r1=x;r1<n;r1++){
for(int l2=y;l2>=0;l2--){
for(int r2=y;r2<n;r2++){
long long sum=0,cnt=0;
for(int i=0;i<n;i++){
long long t=0;
if(l1<=i && i<=r1){
t=d1[i];
cnt++;
}
if(l2<=i && i<=r2){
t=max(t,d2[i]);
cnt++;
}
sum+=t;
}
if(sum<=k){
ans=max(ans,cnt);
}
}
}
}
}
long long cnt=0,sum=0;
vector<long long> v;
for(int i=0;i<n;i++){
v.push_back(d1[i]);
}
sort(v.begin(),v.end());
for(long long x:v){
if(sum+x<=k){
sum+=x;
cnt++;
}
}
ans=max(ans,cnt);
cnt=sum=0;
v.clear();
for(int i=0;i<n;i++){
v.push_back(d2[i]);
}
sort(v.begin(),v.end());
for(long long x:v){
if(sum+x<=k){
sum+=x;
cnt++;
}
}
ans=max(ans,cnt);
return ans;
}
dfs(x,-1,0);
dfs(y,-1,1);
vector<int> a,b;
for(int mask=0;mask<(1<<n);mask++){
for(int i=0;i<n;i++){
A[i]=0;
if((mask&(1<<i))){
A[i]=1;
}
}
ok=0;
dfs1(x,-1,1,mask);
if(!ok){
a.push_back(mask);
}
ok=0;
dfs1(y,-1,1,mask);
if(!ok){
b.push_back(mask);
}
}
int ans=0;
for(int x:a){
for(int y:b){
long long sum=0;
int cnt=0;
for(int i=0;i<n;i++){
long long t=0;
if(((1<<i)&x)){
t=d1[i];
cnt++;
}
if(((1<<i)&y)){
t=max(t,d2[i]);
cnt++;
}
sum+=t;
}
if(sum<=k){
ans=max(ans,cnt);
}
}
}
return ans;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
7772 KB |
2nd lines differ - on the 1st token, expected: '3', found: '4' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
126 ms |
32532 KB |
1st lines differ - on the 1st token, expected: '451', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
7768 KB |
Output is correct |
2 |
Execution timed out |
1064 ms |
16072 KB |
Time limit exceeded |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
7768 KB |
Output is correct |
2 |
Execution timed out |
1064 ms |
16072 KB |
Time limit exceeded |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
7768 KB |
Output is correct |
2 |
Execution timed out |
1064 ms |
16072 KB |
Time limit exceeded |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
7772 KB |
2nd lines differ - on the 1st token, expected: '3', found: '4' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
7772 KB |
2nd lines differ - on the 1st token, expected: '3', found: '4' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
7772 KB |
2nd lines differ - on the 1st token, expected: '3', found: '4' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
7772 KB |
2nd lines differ - on the 1st token, expected: '3', found: '4' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
7772 KB |
2nd lines differ - on the 1st token, expected: '3', found: '4' |
2 |
Halted |
0 ms |
0 KB |
- |