codeforces round#517

链接

A

套上题目里给的公式就过了

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+7;
typedef long long ll;
ll w,h,k,ans;
int main()
{
    scanf("%lld%lld%lld",&w,&h,&k);

    for(int i=1;i<=k;i++)
        ans += 2*((w-4*(i-1))+(h-4*(i-1)))-4;
    printf("%lld\n",ans);

    return 0;
}

B

写了个dp，dp[i][j],，表示第i个位置放j时候的方案数，那么看最后的dp[n-1][j]是否存在就可以了

倒着扫一遍输出路径

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5+7;
typedef long long ll;
int n,flag;
int a[maxn],b[maxn],c[maxn][4];
int ans[maxn];
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n-1;i++)scanf("%d",&a[i]);
    for(int i=0;i<n-1;i++)scanf("%d",&b[i]);
    c[0][0] = c[0][1] = c[0][2] = c[0][3] = 1;
    for(int i=0;i<n-1;i++){
        flag = 0;
        for(int j=0;j<4;j++){
            if(c[i][j]){
                for(int k=0;k<4;k++){
                    if(a[i] == (j|k) && b[i] == (j&k)){
                        c[i+1][k]++,flag++;
                    }
                }
            }
        }
        if(!flag)break;
    }
    if(flag)printf("NO\n");
    else{
        printf("YES\n");
        for(int i=0;i<4;i++){
            if(c[n-1][i]){
                ans[n-1] = i;
                break;
            }
        }
        for(int i=n-2;i>=0;i--){
            for(int j=0;j<4;j++){
                if(c[i][j] && a[i] == (j|ans[i+1]) && b[i] == (j&ans[i+1])){
                    ans[i] = j;
                    break;
                }
            }
        }
        for(int i=0;i<n;i++)printf("%d ",ans[i]);
    }
    return 0;
}

C

我写的很乱，最后wa几发后总算过了，想法是后来看到的（自己的想法太乱，虽然能A，但是决定不说了），

考虑其实很简单，稍微推一下能看出来肯定是个连续的1*****x

(x+1)*x/2=(a+b)，求出x来，倒着扫一遍，给a或给b，最后输出即可

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e7+7;
typedef long long ll;
ll a,b,x,tx,sum;
int i,j,k;
bool ans[maxn];
int main()
{
    cin>>a>>b;
    tx = x = sqrt(2*(a+b)+0.25)-0.5;
    while(x){
        if(a>=x)
            ans[x] = 1,i++,a -= x;
        else
            ans[x] = 0,j++,b -= x;
        x--;
    }
    printf("%d\n",i);
    for(k=1;k<=tx;k++){
        if(ans[k])
            printf("%d ",k);
    }
    printf("\n%d\n",j);
    for(k=1;k<=tx;k++){
        if(!ans[k])
            printf("%d ",k);
    }
    return 0;
}