tql

大佬看看这个第二题的思路对不对，笔试之后听同学问的题目描述

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>


using namespace std;
int n,m,q;
int num[1<<21],num1[1<<21];
long long a[55],b[55];
long long ans;
long long temp,tempcnt;
void merge_sort(int l,int r,int level)
{
    if(l==r) return;
    int mid=(l+r)/2,i,j,k;
    i=l;k=l;j=mid+1;
    merge_sort(l,mid,n-1);
    merge_sort(mid+1,r,n-1);
    while(i<=mid && j<=r)
    {
        if(num[i]<=num[j])
            num1[k++]=num[i++];
        else
        {
            a[level]+=mid-i+1;
            num1[k++]=num[j++];
        }
    }
    while(i<=mid)
        num1[k++]=num[i++];
    while(j<=r)
        num1[k++]=num[j++];
    for(int o=l;o<=r;++o)
        num[o]=num1[o];
}
int main()
{
    scanf("%d",&n);
    temp=1;
    b[0]=0;
    for(int i=1;i<=n;i++)
    {
        temp*=2;
        b[i]=temp*(temp-1)/2;
        for(int j=0;j<i;j++)
        {
            b[j]*=2;
            b[i]-=b[j];
        }
    }
    for(int i=0;i<(1<<n);i++)
        scanf("%d",&num[i]);
    merge_sort(0,(1<<n)-1,n);
    temp=0;
    for(int i=1;i<(1<<n);i++)
    {
        if(num[i]!=num[i-1])
        {
            if(tempcnt!=0)
                temp+=(tempcnt-1)*tempcnt/2;
            tempcnt=0;
        }
        else
        {
            if(tempcnt==0)
                tempcnt=2;
            else
                tempcnt++;
        }
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++)
    {
        scanf("%d",&q);
        for(int j=0;j<=q;j++)
        {
            a[j]=b[j]-a[j];
        }
        ans=0;
        for(int j=0;j<=n;j++)
        {
            ans+=a[j];
        }
        ans-=temp;
        printf("%lld\n",ans);
    }
    return 0;
}