[CF434D]Nanami's Power Plant
题目大意:
(N)个二次函数(f_i(x)=a_ix^2+b_ix+c_i(l_ile xle r_i)),(M)个约束条件((u,v,d)),表示(x_ule x_v+d)。请最大化(sum_{i=1}^nf_i(x_i)),其中(x_i)是整数。
对于(100\%)的数据,(1 leq N leq 50 , 0 leq M leq 100 , |a_i| leq 10 , |b_i| , |c_i| leq 1000 , −100 leq l_i leq r_i leq 100 , 1 leq u , v leq N , u eq v , |d| leq 200),保证所有数均为整数。
思路:
将每个二次函数(i)拆成(node(i,l_i-1)sim node(i,r_i))共(r_i-l_i+2)个点。
对于所有的(node(i,l_i-1)),连边(s o node(i,l_i-1)),容量为(infty)。
对于所有的(node(i,r_i)),连边(s o node(i,r_i)),容量为(infty)。
用(lim)表示一个足够大的数,对于每个二次函数(i)中所有的(node(i,j)),连边(node(i,j-1) o node(i,j)),容量为(lim-f_i(j))。
考虑约束条件((u,v,d)),连边(node(u,j) o node(v,j−d)),容量为(infty)。
答案即为(lim imes n-maxflow)。
源代码:
#include<map>
#include<queue>
#include<cstdio>
#include<cctype>
#include<climits>
#include<cstring>
inline int getint() {
register char ch;
register bool neg=false;
while(!isdigit(ch=getchar())) neg|=ch=='-';
register int x=ch^'0';
while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
return neg?-x:x;
}
const int N=51,V=1e4+1,E=1e5+1;
struct Edge {
int from,to,remain,next;
};
Edge e[E];
int h[V],sz,cnt;
inline void add_edge(const int &u,const int &v,const int &w) {
e[sz]=(Edge){u,v,w,h[u]};h[u]=sz++;
e[sz]=(Edge){v,u,0,h[v]};h[v]=sz++;
}
struct Func {
int a,b,c;
int operator () (const int &x) const {
return a*x*x+b*x+c;
}
};
Func f[N];
int l[N],r[N];
int lev[V],s,t,cur[V];
inline void bfs() {
memset(lev,-1,sizeof lev);
lev[s]=0;
std::queue<int> q;
q.push(s);
while(!q.empty()) {
const int &x=q.front();
for(register int i=h[x];~i;i=e[i].next) {
const int &y=e[i].to,&r=e[i].remain;
if(r&&!~lev[y]) {
lev[y]=lev[x]+1;
q.push(y);
}
}
q.pop();
}
}
int dfs(const int &x,const int &flow) {
if(x==t) return flow;
for(int &i=cur[x];~i;i=e[i].next) {
const int &y=e[i].to;
if(e[i].remain&&lev[y]>lev[x]) {
if(int d=dfs(y,std::min(flow,e[i].remain))) {
e[i].remain-=d;
e[i^1].remain+=d;
return d;
}
}
}
return 0;
}
inline int dinic() {
int maxflow=0;
for(;;) {
bfs();
if(!~lev[t]) break;
memcpy(cur,h,sizeof h);
while(int flow=dfs(s,INT_MAX)) {
maxflow+=flow;
}
}
return maxflow;
}
std::map<int,int> node[N];
int main() {
memset(h,-1,sizeof h);
const int n=getint(),m=getint();
for(register int i=1;i<=n;i++) {
const int a=getint(),b=getint(),c=getint();
f[i]=(Func){a,b,c};
}cnt=1;
s=cnt++;
for(register int i=1;i<=n;i++) {
l[i]=getint(),r[i]=getint();
for(register int j=l[i]-1;j<=r[i];j++) {
node[i][j]=cnt++;
}
}
const int M=INT_MAX/n;
for(register int i=1;i<=n;i++) {
add_edge(s,node[i][l[i]-1],INT_MAX);
for(register int j=l[i];j<=r[i];j++) {
add_edge(node[i][j-1],node[i][j],M-f[i](j));
}
add_edge(node[i][r[i]],t,INT_MAX);
}
for(register int i=0;i<m;i++) {
const int u=getint(),v=getint(),d=getint();
for(register int j=std::max(l[u],l[v]+d)-1;j<=std::min(r[u],r[v]+d);j++) {
add_edge(node[u][j],node[v][j-d],INT_MAX);
}
}
printf("%d
",M*n-dinic());
return 0;
}