#include using namespace std; //Minimum Cost Flow O(FE log V) #include #include #include #include #include template struct MCF{ struct edge{ int to,rev,cap; T cost; }; vector >G; vectorh,d; vectorpv,pe; MCF(int n_=0):G(n_),h(n_),d(n_),pv(n_),pe(n_){} void add_edge(int from,int to,int cap,T cost) { G[from].push_back({to,(int)G[to].size(),cap,cost}); G[to].push_back({from,(int)G[from].size()-1,0,-cost}); } T min_cost_flow(int s,int t,int f)//ans or -1 { fill(h.begin(),h.end(),numeric_limits::max()); h[s]=0; for(int tm=0;tm::max()) { for(int j=0;j0&&h[e.to]>h[i]+e.cost) { h[e.to]=h[i]+e.cost; ch=true; } } } if(!ch)break; if(tm==G.size()-1)return -1;//negative cycle } T ret=0; while(f>0) { fill(d.begin(),d.end(),numeric_limits::max()); d[s]=0; priority_queue,vector >,greater > >P; P.push(make_pair(0,s)); while(!P.empty()) { pairp=P.top();P.pop(); if(d[p.second]0&&d[e.to]>d[p.second]+e.cost+h[p.second]-h[e.to]) { d[e.to]=d[p.second]+e.cost+h[p.second]-h[e.to]; pv[e.to]=p.second; pe[e.to]=i; P.push(make_pair(d[e.to],e.to)); } } } if(d[t]==numeric_limits::max())return -1; for(int u=0;u>N>>K; MCFQ(2*N+2); int st=2*N,go=st+1; for(int i=0;i>A[i]; Q.add_edge(st,i,A[i],0); } for(int i=0;i>B; Q.add_edge(N+i,go,B,0); } for(int i=0;i>P; ans+=P*P; for(int k=1;k<=A[i];k++)Q.add_edge(i,N+j,1,(P-k)*(P-k)-(P-k+1)*(P-k+1)); } cout<