CodeForces 703B(容斥定理)

时间:2023-03-08 20:33:53

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

CodeForces 703B(容斥定理)

解题思路:

第一次写 先求出每个点到其他点的价值,并将其记录 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=;
memset(dp,,sizeof(dp));
for(i=; i<=n; i++)
{
scanf("%I64d",&nar[i]);
sum+=nar[i];
}
long long ans=nar[]*nar[n];
for(int i=; i<n; i++)
ans+=nar[i]*nar[i+];
for(i=; i<=k; i++)
{
scanf("%d",&ka);
long long tem=sum;
if(ka==)
{
if(!dp[])tem-=nar[];
if(!dp[n])tem-=nar[n];
}
else if(ka==n)
{
if(!dp[])tem-=nar[];
if(!dp[n-])tem-=nar[n-];
}
else
{
if(!dp[ka-])tem-=nar[ka-];
if(!dp[ka+])tem-=nar[ka+];
}
ans+=(tem-nar[ka])*nar[ka];
sum-=nar[ka];
dp[ka]=;
}
printf("%I64d\n",ans);
}
return ;
}