QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#609388 | #5507. Investors | QHhee004 | ML | 0ms | 5868kb | C++17 | 2.1kb | 2024-10-04 12:41:30 | 2024-10-04 12:41:30 |
Judging History
answer
#include<iostream>
#include<map>
#include<algorithm>
#include<set>
#include<queue>
using namespace std;
const int inf=1e9+7,N=6005;
int n,m;
int tot,a[N],b[N],f[N][N];
#define lowbit(x) x&-x
class Tree_Array {
private:
int bit[N];
public:
void assign(Tree_Array &t) {
for(int i=1; i<=n; i++)
bit[i]=t.bit[i];
}
void set(int t) {
for(int i=0; i<=n; i++)
bit[i]=t;
}
void add_mn(int x,int t) {
while(x)bit[x]=min(bit[x],t),x-=lowbit(x);
}
int get_mn(int x) {
int ans=inf;
while(x<=n)ans=min(bit[x],ans),x+=lowbit(x);
return ans;
}
void add(int x,int y) {
while(x)bit[x]+=y,x-=lowbit(x);
}
int get(int x) {
int sum=0;
while(x<=n)sum+=bit[x],x+=lowbit(x);
return sum;
}
} bit1,bit2;
void prepare() {
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++) {
scanf("%d",&a[i]);b[i]=a[i];
}
map<int,int>F;
tot=0;
sort(b+1,b+1+n);
for(int i=1; i<=n; i++)
if(F[b[i]]==0)F[b[i]]=++tot;
for(int i=1; i<=n; i++)a[i]=F[a[i]];
}
void Get_f() {
for(int L=1; L<=n; L++) {
bit1.set(0);
bit1.add(a[L],1);
for(int R=L+1; R<=n; R++) {
f[L][R]=f[L][R-1]+bit1.get(a[R]+1);
bit1.add(a[R],1);
}
}
}
struct node {
int now,sum,cnt;
} ;
queue <node> Q;
void Get_ans() {
Q.push(node{1,0,0});
int ans=inf;
bit1.set(inf);
bit2.set(inf);
while(!Q.empty()) {
int x=Q.front().now,res=Q.front().sum,c=Q.front().cnt;
Q.pop();
// if(c==10)printf("c:%d\n",c);
// printf("[%d %d %d]\n",x,res,c);
if(c==m+1&&x==n+1)ans=min(ans,res);
if(c==m+1||x==n+1||res>=bit1.get_mn(x+1))continue;
bit2.add_mn(x,res);
for(int i=x; i<=n; i++) {
// if(i<n&&res+f[x][i]>=bit2.get_mn(i+1))continue;
Q.push(node{i+1,res+f[x][i],c+1});
}
if(Q.front().cnt>c) {
bit1.assign(bit2);
bit2.set(inf);
}
}
printf("%d\n",ans);
}
void solve() {
prepare();
Get_f();
Get_ans();
}
int main() {
// freopen("data.in","r",stdin);
// freopen("my.out","w",stdout);
int T;scanf("%d",&T);
while(T--) solve();
// fclose(stdin);
// fclose(stdout);
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 5868kb
input:
2 6 1 4 5 6 2 2 1 6 2 4 5 6 2 2 1
output:
2 0
result:
ok 2 lines
Test #2:
score: -100
Memory Limit Exceeded
input:
349 6 2 2 1 2 1 2 1 48 12 42 47 39 39 27 25 22 44 45 13 5 48 38 4 37 6 24 10 42 38 12 37 31 19 44 6 29 17 7 12 7 26 35 24 15 9 37 3 27 21 33 29 34 20 14 30 31 21 48 12 1 43 17 46 17 39 40 22 25 2 22 12 4 11 38 12 4 11 1 5 39 44 37 10 19 20 42 45 2 45 37 20 48 34 16 41 23 18 13 44 47 21 29 4 23 18 16...