結果
| 問題 | No.1324 Approximate the Matrix |
| ユーザー |
 kotatsugame
|
| 提出日時 | 2020-12-21 05:57:05 |
| 言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,981 bytes |
| コンパイル時間 | 929 ms |
| コンパイル使用メモリ | 87,192 KB |
| 実行使用メモリ | 300,004 KB |
| 最終ジャッジ日時 | 2024-09-21 12:40:02 |
| 合計ジャッジ時間 | 4,436 ms |
|
ジャッジサーバーID （参考情報） |
judge5 / judge3 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 2 ms
10,624 KB |
| testcase_01 | AC | 2 ms
5,376 KB |
| testcase_02 | AC | 2 ms
5,376 KB |
| testcase_03 | TLE | - |
| testcase_04 | -- | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
| testcase_25 | -- | - |
| testcase_26 | -- | - |
| testcase_27 | -- | - |
| testcase_28 | -- | - |
| testcase_29 | -- | - |
| testcase_30 | -- | - |
| testcase_31 | -- | - |
| testcase_32 | -- | - |
| testcase_33 | -- | - |
| testcase_34 | -- | - |
| testcase_35 | -- | - |
| testcase_36 | -- | - |
| testcase_37 | -- | - |
| testcase_38 | -- | - |
| testcase_39 | -- | - |
| testcase_40 | -- | - |
| testcase_41 | -- | - |
| testcase_42 | -- | - |
| testcase_43 | -- | - |
| testcase_44 | -- | - |
コンパイルメッセージ
main.cpp:68:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
68 | main()
| ^~~~
ソースコード
#include<iostream>
using namespace std;
//Minimum Cost Flow O(FE log V)
#include<algorithm>
#include<utility>
#include<vector>
#include<queue>
#include<limits>
template<typename T>
struct MCF{
struct edge{
int to,rev,cap;
T cost;
};
vector<vector<edge> >G;
vector<T>h,d;
vector<int>pv,pe;
MCF(int n_=0):G(n_),h(n_,0),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
{
T ret=0;
while(f>0)
{
queue<pair<T,int> >P;
fill(d.begin(),d.end(),numeric_limits<T>::max());
d[s]=0;
P.push(make_pair(0,s));
while(!P.empty())
{
pair<T,int>p=P.front();P.pop();
if(d[p.second]<p.first)continue;
for(int i=0;i<G[p.second].size();i++)
{
edge&e=G[p.second][i];
if(e.cap>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<T>::max())return -1;
for(int u=0;u<G.size();u++)h[u]+=d[u];
int d=f;
for(int u=t;u!=s;u=pv[u])d=min(d,G[pv[u]][pe[u]].cap);
f-=d;
ret+=d*h[t];
for(int u=t;u!=s;u=pv[u])
{
G[pv[u]][pe[u]].cap-=d;
G[u][G[pv[u]][pe[u]].rev].cap+=d;
}
}
return ret;
}
};
int N,K;
int A[200],B[200];
int P[200][200];
int ans=0;
main()
{
cin>>N>>K;
for(int i=0;i<N;i++)cin>>A[i];
for(int i=0;i<N;i++)cin>>B[i];
for(int i=0;i<N;i++)for(int j=0;j<N;j++)
{
cin>>P[i][j];
ans+=P[i][j]*P[i][j];
}
MCF<int>Q(N*N+2*N+2);
int st=N*N+2*N,go=st+1;
for(int i=0;i<N;i++)
{
Q.add_edge(st,N*N+i,A[i],0);
Q.add_edge(N*N+N+i,go,B[i],0);
}
for(int i=0;i<N;i++)for(int j=0;j<N;j++)
{
Q.add_edge(N*N+i,i*N+j,K,-2*P[i][j]);
for(int k=0;k<K;k++)Q.add_edge(i*N+j,N*N+N+j,1,2*k+1);
}
cout<<ans+Q.min_cost_flow(st,go,K)<<endl;
}