题解:
首先,这其实是一道最小割的问题,贪心可解纯属巧合。
首先要明确几个性质
1、因为只能连边,无法删边,所以只能使每个点的距离值变小,不可能增大。
2、因为性质1,所以当每一个点的距离值小于期望时,不可能主动改变这个点的距离值。
3、在性质2中,尽管不会主动修改某个距离值,但有可能被迫修改,即修改了它某一条最短路上的结点,那么这个点的距离值也会被迫修改。
所以,我们只需要枚举每一个距离值小于期望值的点,将它的距离值强制减一。
(在实际的图中,只要它不是直接连的原点,一定可以在它的最短路径上找到一个和它距离为2的点,连接这两个点,就可以实现距离值-1)
如果减一后的新结果比原结果小,就可以更新每个点的距离值。
然后重新更新最短路,再从头找一边,直到所有的点都无法更新后,就可以退出。
看到这里,你一定会认为这个方法必然会超时,但其实仔细想想,“将距离值减一”,这一操作看似暴力,但其实最多只会进行n²次(因为只距离值最多为n,且有n个点)。
并且根据图的结构,效率还会提高。估计n=1000都没问题
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define SF scanf
#define PF printf
#define MAXN 110
#define INF 0x3FFFFFFF
using namespace std;
int r[MAXN][MAXN],n;
char s[MAXN];
long long ans,ans1,want[MAXN],dist[MAXN],dist1[MAXN];
pair<int,int> tru_d[MAXN];
long long sq(long long x){
return x*x;
}
void prepare_by_sort(){
for(int i=1;i<=n;i++)
tru_d[i]=make_pair(dist[i],i);
sort(tru_d+1,tru_d+n+1);
}
void bfs(){
queue<pair<int,int> > q;
q.push(make_pair(1,0));
dist[1]=0;
while(!q.empty()){
pair<int,int> now=q.front();
q.pop();
int x=now.first,y=now.second;
for(int i=1;i<=n;i++)
if(r[x][i]!=0&&dist[i]==INF){
dist[i]=y+1;
q.push(make_pair(i,y+1));
}
}
}
int main(){
freopen("newroad.in","r",stdin);
freopen("newroad.out","w",stdout);
SF("%d",&n);
for(int i=1;i<=n;i++)
dist[i]=INF;
for(int i=1;i<=n;i++){
SF("%s",s+1);
for(int j=1;j<=n;j++)
if(s[j]=='Y')
r[i][j]=1;
}
for(int i=1;i<=n;i++)
SF("%d",&want[i]);
bfs();
for(int i=1;i<=n;i++)
ans+=sq(want[i]-dist[i]);
bool changed=1;
while(changed==1){
changed=0;
prepare_by_sort();
for(int i=n;i<=n&&tru_d[i].first>=2;i--){
int x=tru_d[i].second;
if(want[x]<tru_d[i].first){
memcpy(dist1,dist,sizeof dist);
dist1[x]--;
int mod=1;
while(mod==1){
mod=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(dist1[i]>dist1[j]+1&&r[j][i]==1){
dist1[i]=dist1[j]+1;
mod=1;
}
}
ans1=0;
for(int i=1;i<=n;i++)
ans1+=sq(dist1[i]-want[i]);
if(ans1<ans){
ans=ans1;
memcpy(dist,dist1,sizeof dist1);
changed=1;
break;
}
}
}
}
PF("%lld",ans);
}