제출 #1359840

#제출 시각아이디문제언어결과실행 시간메모리
1359840minhtienDžumbus (COCI19_dzumbus)C++20
0 / 110
20 ms28212 KiB
#include <bits/stdc++.h>
#define ll long long
#define ii pair<ll,ll>
#define fi first
#define se second
using namespace std;
const int N=1e3+5;
const ll inf=1e18+7;
int n,m;
vector<int>v[N];
ll dp[N][4][N];
ll dp2[N][N];
ll dp3[N];
int sz[N];
vector<ll>v2;
int vs[N];
int a[N];
void dfs(int s,int pr){
    sz[s]=1;
    vs[s]=1;
    dp[s][0][0]=0;
    dp[s][1][0]=a[s];
    for(auto x:v[s]){
        if(x!=pr){
            dfs(x,s);
            int sz1=sz[s]+sz[x];
            ll dp1[4][sz1+1];
            for(int i=0;i<=sz1;i++){
                for(int j=0;j<=2;j++){
                    dp1[j][i]=inf;
                }
            }
            for(int i=0;i<=sz[s];i++){
                for(int j=0;j<=sz[x];j++){
                    dp1[0][i+j]=min(dp1[0][i+j],min(dp[s][0][i]+dp[x][0][j],dp[s][0][i]+dp[x][2][j]));
                    dp1[1][i+j]=min(dp1[1][i+j],min(dp[s][1][i]+dp[x][1][j],dp[s][1][i]+dp[x][0][j]));
                    dp1[2][i+j+1]=min(dp1[2][i+j+1],dp[s][2][i]+dp[x][1][j]);
                    dp1[2][i+j]=min(dp1[2][i+j],dp[s][2][i]+dp[x][2][j]);
                    if(i+j+1<=sz1){
                        dp1[2][i+j+1]=min(dp1[2][i+j+1],dp[s][2][i]+dp[x][1][j]);
                        dp1[2][i+j+1]=min(dp1[2][i+j+1],dp[s][1][i]+dp[x][2][j]);
                    }
                    if(i+j+2<=sz1){
                        dp1[2][i+j+2]=min(dp1[2][i+j+2],dp[s][1][i]+dp[x][1][j]);
                    }
                }
            }
//            cout << " " << s << " " << x << " " << sz1 << " " << dp1[1][3] << " " << dp[s][1][2] << " " << dp[x][1][1]<< "\n";
            for(int i=0;i<=sz1;i++){
                dp[s][0][i]=dp1[0][i];
                dp[s][1][i]=dp1[1][i];
                dp[s][2][i]=dp1[2][i];
            }
            sz[s]+=sz[x];
        }
    }
}
int main()
{
    ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
    cin >>n >>m;
    for(int i=1;i<=n;i++){
        cin >> a[i];
    }
    for(int i=1;i<=m;i++){
        int x,y;
        cin >>x >>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<=2;j++){
            for(int k=1;k<=n;k++){
                dp[i][j][k]=inf;
            }
        }
    }
    for(int i=0;i<=n;i++){
        dp3[i]=inf;
    }
    dp3[0]=0;
    for(int i=1;i<=n;i++){
        if(vs[i]==0){
            dfs(i,-1);
            ll dp4[n+1];
            for(int j=0;j<=n;j++){
                dp4[j]=inf;
            }
            for(int j=0;j<=n;j++){
                for(int k=0;k<=sz[i];k++){
                    if(j+k<=n){
                        dp4[j+k]=min(dp4[j+k],dp3[j]+min(dp[i][0][k],min(dp[i][1][k],dp[i][2][k])));
                    }
                }
            }
            for(int j=0;j<=n;j++){
                dp3[j]=dp4[j];
            }
        }
    }
    for(int i=0;i<=n;i++){
        v2.push_back(dp3[i]);
    }
    int q;
    cin >>q;
//    for(auto x:v2){
//        cout << x << " ";
//    }
//    cout << "\n";
    while(q--){
        ll s;
        cin >>s;
        int t1=upper_bound(v2.begin(),v2.end(),s)-v2.begin();
        t1--;
        if(t1<=1) cout << 0 << "\n";
        else cout << t1 << "\n";
    }
    return 0;
}

/*


3 2
1 2 3
1 2
1 3
3
2
3
5




10 9
99 1 99 99 99 99 99 4 2 3
1 2
1 3
3 4
3 5
4 6
4 7
4 8
7 9
7 10





14 13
2 3 4 19 20 21 5 22 6 7 23 8 10 14
1 2
1 3
1 4
2 5
2 6
3 7
3 8
3 9
4 10
8 11
10 13
10 12
12 14
3
45
44
23

*/
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…