Codeforces - Four Melodies

题目链接:Codeforces - Four Melodies


显然,暴力连边是 n*n 的,肯定会TLE。

考虑优化:我们考虑对%7的值分块,那么同一个块中前面的点可以到后面的所有点,那么我们每个点就只需要连后面的满足的点了。

怎么满足同一个块中前面的可以到后面呢?我们加一条 i->i+n流量为INF,费用为0的边即可。


AC代码:

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
//#define int long long
using namespace std;
const int inf=0x3f3f3f3f;
const int N=1e4+10,M=1e6+10;
int n,a[N],s,s1,t,d[N],st[N],vis[N],cost;
int head[N],nex[M],to[M],w[M],flow[M],tot=1;
inline void ade(int a,int b,int c,int d){
	to[++tot]=b; nex[tot]=head[a]; w[tot]=d; flow[tot]=c; head[a]=tot;
}
inline void add(int a,int b,int c,int d){ade(a,b,c,d);	ade(b,a,0,-d);}
inline int spfa(){
	queue<int> q; q.push(s); memset(st,0,sizeof st); memset(d,-inf,sizeof d); d[s]=0;
	while(q.size()){
		int u=q.front();	q.pop();	vis[u]=0;
		for(int i=head[u];i;i=nex[i]){
			if(flow[i]&&d[to[i]]<d[u]+w[i]){
				d[to[i]]=d[u]+w[i];
				if(!vis[to[i]])	q.push(to[i]),vis[to[i]]=1;
			}
		}
	}
	return d[t]>0;
}
int dfs(int x,int f){
	if(x==t)	return cost+=d[t]*f,f;
	int fl=0;	st[x]=1;
	for(int i=head[x];i&&f;i=nex[i]){
		if(flow[i]&&!st[to[i]]&&d[to[i]]==d[x]+w[i]){
			int mi=dfs(to[i],min(flow[i],f));
			flow[i]-=mi,flow[i^1]+=mi,fl+=mi,f-=mi;
		}
	}
	return fl;
}
signed main(){
	cin>>n;	t=n*2+1; s1=t+1; add(s,s1,4,0);
	for(int i=1;i<=n;i++)	cin>>a[i],add(i,i+n,1,1),add(i,i+n,inf,0);
	for(int i=1;i<=n;i++)	add(s1,i,1,0),add(i+n,t,1,0);
	for(int i=1;i<=n;i++){
		for(int j=i+1;j<=n;j++)	if(a[j]==a[i]-1){
			add(i+n,j,inf,0);	break;
		}
		for(int j=i+1;j<=n;j++)	if(a[j]==a[i]+1){
			add(i+n,j,inf,0);	break;
		}
		for(int j=i+1;j<=n;j++)	if(a[j]%7==a[i]%7){
			add(i+n,j,inf,0);	break;
		}
	}
	while(spfa())	dfs(s,inf);
	cout<<cost;
	return 0;
}
发布了725 篇原创文章 · 获赞 244 · 访问量 4万+

猜你喜欢

转载自blog.csdn.net/weixin_43826249/article/details/104732125