oj.uz

답안 #602917

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
602917 2022-07-23T12:38:15 Z berr Kpart (eJOI21_kpart) C++17
30 / 100
 2000 ms 130760 KB 
#include <bits/stdc++.h>
using namespace std;
int knp[1005][100015];
 
int32_t main()
{
    ios_base::sync_with_stdio(false); cin.tie(0);
    int t; cin>>t; 
    int p=0;
    while(t--)
    {
 
        int n; cin>>n;
        vector<int> pre(n), a(n);
        for(int i=0; i<n; i++)
        {
            cin>>a[i]; 
            if(i==0) pre[i]=a[i];
            else pre[i]=pre[i-1]+a[i];
        }
 
 
 
        for(int i=0; i<=pre[n-1]; i++) knp[0][i]=-1;
        knp[0][a[0]]=0;
 
        knp[0][0]=0; 
        for(int i=1; i<n; i++)
        {
            for(int l=pre[n-1]-a[i]; l>0; l--)
            {
                knp[i][a[i]+l] = max(knp[i-1][a[i]+l], knp[i-1][l]);

            }
            knp[i][a[i]]=i; 
        
            for (int j = a[i]-1; j > 0; j--) knp[i][j] = knp[i - 1][j];
 
        }
    

        vector<int> ans;
        for (int i=2; i<=n; i++)
        {
            int flag=1;
            for (int j=0; j<=n-i; j++)
            {
                int sum = pre[j+i-1];
                if(j-1>=0) sum-=pre[j-1];
            
                if (sum%2==1||knp[j+i-1][sum/2]<j) flag = 0;
            }
            if (flag == 1) ans.push_back(i);
        }

        cout<<ans.size()<<" ";
        for(auto i: ans) cout<<i<<" ";
            cout<<"\n";
 
    }
   
}

Compilation message

Main.cpp: In function 'int32_t main()':
Main.cpp:9:9: warning: unused variable 'p' [-Wunused-variable]
    9 |     int p=0;
      |         ^

Subtask #1
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 768 KB Output is correct

Subtask #2
20.0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Correct 21 ms 2608 KB Output is correct
2 Correct 92 ms 7012 KB Output is correct

Subtask #3
0 / 70.0

# 결과 실행 시간 메모리 Grader output
1 Correct 357 ms 27256 KB Output is correct
2 Correct 916 ms 61404 KB Output is correct
3 Correct 1156 ms 78244 KB Output is correct
4 Execution timed out 2027 ms 130760 KB Time limit exceeded
5 Halted 0 ms 0 KB -