結果

問題 No.957 植林
ユーザー TqkTqk
提出日時 2021-06-14 10:00:18
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 886 ms / 2,000 ms
コード長 11,141 bytes
コンパイル時間 3,761 ms
コンパイル使用メモリ 227,460 KB
実行使用メモリ 22,936 KB
最終ジャッジ日時 2024-12-24 05:03:49
合計ジャッジ時間 25,221 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 136 ms
21,912 KB
testcase_04 AC 123 ms
21,668 KB
testcase_05 AC 121 ms
21,816 KB
testcase_06 AC 162 ms
22,432 KB
testcase_07 AC 139 ms
21,832 KB
testcase_08 AC 87 ms
21,924 KB
testcase_09 AC 82 ms
22,008 KB
testcase_10 AC 94 ms
22,136 KB
testcase_11 AC 90 ms
22,048 KB
testcase_12 AC 86 ms
22,268 KB
testcase_13 AC 67 ms
21,976 KB
testcase_14 AC 82 ms
22,068 KB
testcase_15 AC 77 ms
21,968 KB
testcase_16 AC 68 ms
21,592 KB
testcase_17 AC 70 ms
22,024 KB
testcase_18 AC 639 ms
21,928 KB
testcase_19 AC 668 ms
21,968 KB
testcase_20 AC 709 ms
22,148 KB
testcase_21 AC 734 ms
22,192 KB
testcase_22 AC 765 ms
22,252 KB
testcase_23 AC 795 ms
22,312 KB
testcase_24 AC 832 ms
22,656 KB
testcase_25 AC 886 ms
22,932 KB
testcase_26 AC 867 ms
22,932 KB
testcase_27 AC 882 ms
22,800 KB
testcase_28 AC 870 ms
22,808 KB
testcase_29 AC 862 ms
22,932 KB
testcase_30 AC 862 ms
22,936 KB
testcase_31 AC 586 ms
21,672 KB
testcase_32 AC 620 ms
21,944 KB
testcase_33 AC 638 ms
22,296 KB
testcase_34 AC 643 ms
22,300 KB
testcase_35 AC 691 ms
22,376 KB
testcase_36 AC 730 ms
22,436 KB
testcase_37 AC 723 ms
22,784 KB
testcase_38 AC 754 ms
22,936 KB
testcase_39 AC 770 ms
22,936 KB
testcase_40 AC 854 ms
22,808 KB
testcase_41 AC 43 ms
22,928 KB
testcase_42 AC 41 ms
22,804 KB
testcase_43 AC 69 ms
22,932 KB
testcase_44 AC 78 ms
22,932 KB
testcase_45 AC 2 ms
5,248 KB
testcase_46 AC 2 ms
5,248 KB
testcase_47 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:1:13: warning: extra tokens at end of #ifdef directive
    1 | #ifdef MAIN ================================================================================================================================
      |             ^~
main.cpp:30:8: warning: extra tokens at end of #endif directive [-Wendif-labels]
   30 | #endif MAIN ================================================================================================================================
      |        ^~~~
main.cpp:59:13: warning: integer suffix 'z' shadowed by implementation
   59 | inline lint operator""z(const unsigned long long n){ return lint(n); }
      |             ^~~~~~~~

ソースコード

diff # 

#ifdef MAIN ================================================================================================================================

+library{ name:dinic_notmine }

int solve(){
	lin(h, w);
	vector<vector<lint>> g(h, vector<lint>(w));
	rep(h)in(g[i]);
	vin(lint, r, h);
	vin(lint, c, w);
	each(r)e *= 2;
	each(c)e *= 2;
	Dinic<long long> d(h+w+2);
	rep(h)rep(j, w){
		d.add_edge(0, 1+i, g[i][j]);
		d.add_edge(0, 1+h+j, g[i][j]);
		d.add_edge(1+i, 1+h+j, g[i][j]);
		d.add_edge(1+h+j, 1+i, g[i][j]);
	}
	rep(h)d.add_edge(1+i, 1+h+w, r[i]);
	rep(j, w)d.add_edge(1+h+j, 1+h+w, c[j]);
	lint ans = 0;
	each(r)ans += e;
	each(c)ans += e;
	ans -= d.max_flow(0, 1+h+w);
	out(ans/2);
	return 0;
}

#endif MAIN ================================================================================================================================

//sub-BOF

// desktop
// author: Tqk

// #define _AOJ_
#define _C_INPUT_

// #pragma region template

// #pragma region basic
#define IN_FILE "./in.txt"
//#pragma GCC optimize ("O3")
#pragma warning(disable: 4455 4244 4067 4068 4996)
#pragma GCC target ("avx")
#pragma GCC diagnostic ignored "-Wliteral-suffix"
#define NOMINMAX
#ifdef _ENV_TQK_
#include <Windows.h>
#endif
#include "bits/stdc++.h"
using namespace std;
typedef int64_t lint;
typedef long double ld;
typedef string cs;
#define linf 1152921504606846976
#ifndef _AOJ_
inline lint operator""z(const unsigned long long n){ return lint(n); }
#endif
template<class T>inline T operator-(const vector<T>& a,const size_t&i){return a.at(i);}
// #pragma endregion

// #pragma region rep
#define _vcppunko3(tuple) _getname3 tuple
#define _vcppunko4(tuple) _getname4 tuple
#define _getname4(_1,_2,_3,_4,name,...) name
#define _getname3(_1,_2,_3,name,...) name
#define _trep2(tuple) _rep2 tuple
#define _trep3(tuple) _rep3 tuple
#define _trep4(tuple) _rep4 tuple
#define _rep1(n) for(lint i=0;i<n;++i)
#define _rep2(i,n) for(lint i=0;i<n;++i)
#define _rep3(i,a,b) for(lint i=a;i<b;++i)
#define _rep4(i,a,b,c) for(lint i=a;i<b;i+=c)
#define _tper2(tuple) _per2 tuple
#define _tper3(tuple) _per3 tuple
#define _tper4(tuple) _per4 tuple
#define _per1(n) for(lint i=n-1;i>=0;--i)
#define _per2(i,n) for(lint i=n-1;i>=0;--i)
#define _per3(i,a,b) for(lint i=b-1;i>=a;--i)
#define _per4(i,a,b,c) for(lint i=a+(b-a-1)/c*c;i>=a;i-=c)
#define rep(...) _vcppunko4((__VA_ARGS__,_trep4,_trep3,_trep2,_rep1))((__VA_ARGS__))
#define per(...) _vcppunko4((__VA_ARGS__,_tper4,_tper3,_tper2,_per1))((__VA_ARGS__))
#define _tall2(tuple) _all2 tuple
#define _tall3(tuple) _all3 tuple
#define _all1(v) v.begin(),v.end()
#define _all2(v,a) v.begin(),v.begin()+a
#define _all3(v,a,b) v.begin()+a,v.begin+b
#define all(...) _vcppunko3((__VA_ARGS__,_tall3,_tall2,_all1))((__VA_ARGS__))
#define each(c) for(auto &e:c)
// #pragma endregion

// #pragma region io
template<class T>
istream& operator>>(istream& is,vector<T>& vec);
template<class T,size_t size>
istream& operator>>(istream& is,array<T,size>& vec);
template<class T,class L>
istream& operator>>(istream& is,pair<T,L>& p);
template<class T>
ostream& operator<<(ostream& os,vector<T>& vec);
template<class T,class L>
ostream& operator<<(ostream& os,pair<T,L>& p);
template<class T>
istream& operator>>(istream& is,vector<T>& vec){ for(T& x: vec) is>>x;return is; }
template<class T,class L>
istream& operator>>(istream& is,pair<T,L>& p){ is>>p.first;is>>p.second;return is; }
template<class T,class L>
ostream& operator<<(ostream& os,pair<T,L>& p){ os<<p.first<<" "<<p.second;return os; }
template<class T>
ostream& operator<<(ostream& os,vector<T>& vec){ if(vec.size())os<<vec[0];rep(i,1,vec.size())os<<' '<<vec[i];return os; }
template<class T>
ostream& operator<<(ostream& os,deque<T>& deq){ if(deq.size())os<<deq[0];rep(i,1,deq.size())os<<' '<<deq[i];return os; }
template<class T>
ostream& operator<<(ostream& os,set<T>& s){ each(s)os<<e<<" ";return os; }

#ifdef _ENV_TQK_
/*
ifstream infile(IN_FILE);
template<class T>
ifstream& operator>>(ifstream& is,vector<T>& vec);
template<class T,size_t size>
ifstream& operator>>(ifstream& is,array<T,size>& vec);
template<class T,class L>
ifstream& operator>>(ifstream& is,pair<T,L>& p);
template<class T>
ifstream& operator>>(ifstream& is,vector<T>& vec){ for(T& x: vec) is>>x;return is; }
template<class T,class L>
ifstream& operator>>(ifstream& is,pair<T,L>& p){ is>>p.first;is>>p.second;return is; }
inline void fin(){}
template <class Head,class... Tail>
inline void fin(Head&& head,Tail&&... tail){ infile>>head;fin(move(tail)...); }
*/
//#include<Windows.h>
HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
inline void in(){ SetConsoleTextAttribute(hConsole,10); }
template <class Head,class... Tail>
inline void in(Head&& head,Tail&&... tail){
	SetConsoleTextAttribute(hConsole,15);
	cin>>head;in(move(tail)...);
}
#else
inline void in(){}
template <class Head,class... Tail>
inline void in(Head&& head,Tail&&... tail){ cin>>head;in(move(tail)...); }
#endif

inline bool out(){ return(cout<<'\n',0); }
template <class T>
inline bool out(T t){ return(cout<<t<<'\n',0); }
template <class Head,class... Tail>
inline bool out(Head head,Tail... tail){ cout<<head<<' ';out(move(tail)...);return 0; }
template <class T>
inline void alloc(T &c,lint s){ rep(c.size())c[i].resize(s); }
#define alc alloc
#ifdef _ENV_TQK_
inline bool deb(){ SetConsoleTextAttribute(hConsole,10); return(cout<<'\n',0); }
template <class T>
inline bool deb(T t){ return(SetConsoleTextAttribute(hConsole,12),cout<<t<<'\n',SetConsoleTextAttribute(hConsole,10),0); }
template <class Head,class... Tail>
inline bool deb(Head head,Tail... tail){
	SetConsoleTextAttribute(hConsole,12);
	cout<<head<<' ';deb(move(tail)...);return 0;
}
#define dsp(ex) sp(ex)
#define dno(ex) no(ex)
#define look(v) SetConsoleTextAttribute(hConsole,12),cout<<#v<<": ",deb(v);
#else
#define deb(...) 0
#define dsp(ex) 0
#define dno(ex) 0
#define look(v) 0
#endif

// #pragma endregion

// #pragma region TA
#define lin(...) lint __VA_ARGS__;in(__VA_ARGS__)
#define stin(...) string __VA_ARGS__;in(__VA_ARGS__)
#define vin(type,name,size) vector<type> name(size);in(name)
#define pb push_back
#define fi e.first
#define se e.second
#define YES(c) cout<<((c)?"YES\n":"NO\n"),0
#define Yes(c) cout<<((c)?"Yes\n":"No\n"),0
#define POSSIBLE(c) cout<<((c)?"POSSIBLE\n":"IMPOSSIBLE\n"),0
#define Possible(c) cout<<((c)?"Possible\n":"Impossible\n"),0
#define o(p) cout<<p<<endl,0
#define sp(p) cout<<p<<" ",0
#define no(p) cout<<p,0
#define st(v) sort(all(v))
#define psort(l,r) if(l>r)swap(l,r);
#define fn(ty1,ty2,ex) [&](ty1 l,ty2 r){ return(ex); }
#define lfn(ex) [&](lint l,lint r){ return(ex); }

// #pragma endregion

// #pragma region func
inline constexpr lint gcd(lint a,lint b){ while(b){ lint c=b;b=a%b;a=c; }return a; }
inline constexpr lint lcm(lint a,lint b){ return a/gcd(a,b)*b; }
template<typename T>
inline constexpr bool chmin(T &mn,const T &cnt){ if(mn>cnt){ mn=cnt;return 1; } else return 0; }
template<typename T>
inline constexpr bool chmax(T &mx,const T &cnt){ if(mx<cnt){ mx=cnt;return 1; } else return 0; }
template <class F>
inline void srt(F f){  }
template <class F,class Head,class... Tail>
inline void srt(F f,Head&& head,Tail&&... tail){
	vector<int>a(head.size());
	iota(all(a),0);
	sort(all(a),f);
	auto res=head;
	rep(head.size())res[i]=head[a[i]];
	head=res;
	srt(f,move(tail)...);
}
auto smaller(vector<lint>&a){
	return [&](int i,int j){return a[i]<a[j];};
}
auto larger(vector<lint>&a){
	return [&](int i,int j){return a[i]>a[j];};
}
// #pragma endregion

// #pragma region P

class P{ public:lint f,s; P(lint a,lint b):f(a),s(b){}; P():f(0),s(0){}; };
istream& operator>>(istream& os,P& p){ os>>p.f>>p.s;return os; }
inline constexpr bool operator<(const P& l,const P& r){ return(l.f-r.f?l.f<r.f:l.s<r.s); }
inline constexpr bool operator>(const P& l,const P& r){ return(l.f-r.f?l.f>r.f:l.s>r.s); }
inline constexpr bool operator==(const P& l,const P& r){ return(l.f==r.f&&l.s==r.s); }
inline constexpr bool operator!=(const P& l,const P& r){ return(l.f!=r.f||l.s!=r.s); }
inline P operator+(const P& l,const P& r){return P(l.f+r.f,l.s+r.s);}
inline P operator-(const P& l,const P& r){return P(l.f-r.f,l.s-r.s);}
inline P operator-(const P& r){return P(-r.f,-r.s); }
inline P operator*(const lint& l,const P& r){return P(l*r.f,l*r.s); }
inline P operator/(const P& l,const lint& r){return P(l.f/r,l.s/r); }
inline ld abs(P p){return sqrtl(p.f*p.f+p.s*p.s);}
inline ld mabs(P p){return abs(p.f)+abs(p.s);}
inline ld mht(P x,P y){return mabs(x-y);}

// #pragma endregion

// #pragma region start
struct Start{
	Start(){
#ifndef _C_INPUT_
		cin.tie(0),cout.tie(0);
		ios::sync_with_stdio(0);
#endif
		cout<<fixed<<setprecision(16);
	}
} __start;
// #pragma endregion

// #pragma endregion

// #pragma region const
#define mod 1000000007
const ld pi=acos(-1);
const ld tau=(1.+sqrtl(5))/2.;
P d4[4]={P(1,0),P(0,1),P(-1,0),P(0,-1)};
P d8[8]={P(1,0),P(1,1),P(0,1),P(-1,1),P(-1,0),P(-1,-1),P(0,-1),P(1,-1)};
const cs AUTO_YES = "Yes";
const cs AUTO_NO = "No";
int cho(bool c,cs yes=AUTO_YES,cs no=AUTO_NO){
	return out((c?yes:no));
}
// #pragma endregion

// #pragma region solve

// #pragma region lib_dinic_notmine

template< typename T >
struct Dinic
{
	struct edge
	{
		int to;
		T cap;
		int rev;
	};

	const T INF;
	vector< vector< edge > > graph;
	vector< T > min_cost;
	vector< int > iter;

	Dinic(int n): INF(numeric_limits< T >::max())
	{
		graph.resize(n);
	}

	void add_edge(int from, int to, T cap)
	{
		graph[from].push_back({ to, cap, (int)graph[to].size() });
		graph[to].push_back({ from, 0, (int)graph[from].size()-1 });
	}

	bool bfs(int s, int t)
	{
		min_cost.assign(graph.size(), -1);
		queue< int > que;
		min_cost[s] = 0;
		que.push(s);
		while(!que.empty()) {
			int p = que.front();
			que.pop();
			for(auto& e:graph[p]) {
				if(e.cap>0&&min_cost[e.to]==-1) {
					min_cost[e.to] = min_cost[p]+1;
					que.push(e.to);
				}
			}
		}
		return (min_cost[t]!=-1);
	}

	T dfs(int idx, const int t, T flow)
	{
		if(idx==t) return (flow);
		for(int& i = iter[idx]; i<graph[idx].size(); i++) {
			edge& e = graph[idx][i];
			if(e.cap>0&&min_cost[idx]<min_cost[e.to]) {
				T d = dfs(e.to, t, min(flow, e.cap));
				if(d>0) {
					e.cap -= d;
					graph[e.to][e.rev].cap += d;
					return (d);
				}
			}
		}
		return (0);
	}

	T max_flow(int s, int t)
	{
		T flow = 0;
		while(bfs(s, t)) {
			iter.assign(graph.size(), 0);
			T f = 0;
			while((f = dfs(s, t, INF))>0) {
				flow += f;
			}
		}
		return (flow);
	}
};

// #pragma endregion

int solve(){
	lin(h, w);
	vector<vector<lint>> g(h, vector<lint>(w));
	rep(h)in(g[i]);
	vin(lint, r, h);
	vin(lint, c, w);
	each(r)e *= 2;
	each(c)e *= 2;
	Dinic<long long> d(h+w+2);
	rep(h)rep(j, w){
		d.add_edge(0, 1+i, g[i][j]);
		d.add_edge(0, 1+h+j, g[i][j]);
		d.add_edge(1+i, 1+h+j, g[i][j]);
		d.add_edge(1+h+j, 1+i, g[i][j]);
	}
	rep(h)d.add_edge(1+i, 1+h+w, r[i]);
	rep(j, w)d.add_edge(1+h+j, 1+h+w, c[j]);
	lint ans = 0;
	each(r)ans += e;
	each(c)ans += e;
	ans -= d.max_flow(0, 1+h+w);
	out(ans/2);
	return 0;
}

// #pragma endregion

// #pragma region main

int main(){
	solve();
}

// #pragma endregion

//sub-EOF
0