# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
602655 | berr | Kpart (eJOI21_kpart) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
int32_t main()
{
ios_base::sync_with_stdio(false); cin.tie(0);
int t; cin>>t;
while(t--)
{
int n; cin>>n;
vector<int> pre(n), a(n), count(n+1);
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];
}
vector<vector<int>> check(n, vector<int>(pre[n-1]+5, 0));
vector<int> knp(pre[n-1]+5, -1);
knp[0]=0;
for(int i=0; i<n; i++)
{
for(int l=2; l<=i+1; l++)
{
int a=pre[i];
if(i-l>=0) a-=pre[i-l];
if(a%2==0)
{
check[i][a/2]=l;
}
}
}
count[n]=1;
for(int i=n-1; i>=0; i--)
{
count[i]=count[i+1]+1;
}
vector<int> q;
vector<int> vis(100005);
for(int i=0; i<n; i++)
{
int z=q.size();
for(int l=0; l<z; l++)
{
if(vis[q[l]+a[i]]==0)
{
q.push_back(q[l]+a[i]);
vis[q[l]+a[i]]=1;
}
if(check[i][q[l]+a[i]]&&max(knp[q[l]], knp[q[l]])+a[i]])>=i-check[i][q[l]+a[i]]+1)
{
count[check[i][q[l]+a[i]]]--;
}
}
knp[a[i]]=i;
if(check[i][a[i]])
{
count[check[i][a[i]]]--;
}
if(vis[a[i]]==0)
{
vis[a[i]]=1;
q.push_back(a[i]);
}
for(int l=0; l<z; l++)
{
knp[q[l]+a[i]]=max(knp[q[l]+a[i]], knp[q[l]]);
}
knp[a[i]]=i;
// sort(q.begin(), q.end(), greater<int>());
}
vector<int> ans;
for(int i=2; i<=n; i++)
{
if(count[i]==0) ans.push_back(i);
}
cout<<ans.size()<<" ";
for(auto i: ans) cout<<i<<" ";
cout<<"\n";
}
}