題意：有n座城市，其中k座是省會城市，每個城市有對應的點權，城市1-2-3-...-n-1有一條路相連，省會城市與其他所有的城市相連，且每兩個城市間最多有一條路，每條路的邊權為路連接的兩座城市的點權乘積，問所有路的邊權和。

思路：預處理出所有城市的點權和與所有省會城市的點權和，對于每座城市，如果它不是省會城市的話，它只與它的前一個城市與后一個城市以及所有的省會城市有連邊，如果它是省會城市的話，它與除了自己的所有城市有連邊。這樣每條邊在邊的兩個點處都考慮了一次，答案/2。

?

#include<cstdio> int val[100005]; int cab[100005]; typedef long long ll; int main() {int n,k,x;ll city_sum=0,cab_sum=0;scanf("%d%d",&n,&k);for(int i=1;i<=n;i++) {scanf("%d",&val[i]);city_sum+=val[i];}for(int i=1;i<=k;i++) {scanf("%d",&x);cab[x]=1;cab_sum+=val[x];}ll res=0;for(int i=1;i<=n;i++) {if(cab[i]) {res+=(city_sum-val[i])*val[i];}else {ll temp=cab_sum;if(i==1) {if(!cab[n]) temp+=val[n];if(!cab[2]) temp+=val[2];}else if(i==n) {if(!cab[n-1]) temp+=val[n-1];if(!cab[1]) temp+=val[1];}else {if(!cab[i-1]) temp+=val[i-1];if(!cab[i+1])temp+=val[i+1];}res+=temp*val[i];}}printf("%I64d\n",res/2);return 0; }

?

轉載于:https://www.cnblogs.com/LinesYao/p/5742962.html

總結

以上是生活随笔為你收集整理的codeforces 703B的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。