CodeForces 703B(容斥定理)

题目链接：http://codeforces.com/contest/703/problem/B

解题思路：

第一次写 先求出每个点到其他点的价值，并将其记录 dp[i][j]=1（i<j），然后算出周围一圈的价值，当然有dp[i][j]来防止重复计算，超时

第二次写 将二维数组用一维数组代替 方法是 dp[i][j]=dp[i*10+j] (i<j); 然后求出一圈的价值，dp[i] 来防止重复计算，超时

第三次写 先求出所有点价值的和sum，以及一圈的价值ans，如果 省会 k==1 则在总和中减去前一个和后一个，即减去第2个和第n个

ans+=（tem-a[1]）*a[1];

sum-=a[1];

dp[x]=1;//防止重复计算

   关键点是找到前一个和后一个，然后计算到ans中，同时sum-=当前值，记得dp[x]=1;

第一个和最后一个特殊处理。

Ac code：

#include<bits/stdc++.h>
using namespace std;
#define maxx 200002
int dp[maxx];
long long nar[maxx];
int main()
{
    int n,k,ka,i;
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        long long sum=0;
        memset(dp,0,sizeof(dp));
        for(i=1; i<=n; i++)
        {
            scanf("%I64d",&nar[i]);
            sum+=nar[i];
        }
        long long  ans=nar[1]*nar[n];
        for(int i=1; i<n; i++)
            ans+=nar[i]*nar[i+1];
        for(i=1; i<=k; i++)
        {
            scanf("%d",&ka);
            long long tem=sum;
            if(ka==1)
            {
                if(!dp[2])tem-=nar[2];
                if(!dp[n])tem-=nar[n];
            }
            else if(ka==n)
            {
                if(!dp[1])tem-=nar[1];
                if(!dp[n-1])tem-=nar[n-1];
            }
            else
            {
                if(!dp[ka-1])tem-=nar[ka-1];
                if(!dp[ka+1])tem-=nar[ka+1];
            }
            ans+=(tem-nar[ka])*nar[ka];
            sum-=nar[ka];
            dp[ka]=1;
        }
        printf("%I64d
",ans);
    }
    return 0;
}