codeforces 1017D The Wu(狀壓+預處理)

傳送門:codeforces 1017D

題意:給你n m q,表示在這一組數據中所有的01串長度均爲n,然後給你一個含有m個元素的multiset,之後有q次詢問。每次詢問會給你一個01串t和一個給定常數k,讓你輸出串t和multiset裏面多少個元素的“Wu”值不超過k。對於“Wu”值的定義:如果兩個01串s和t在位置i上滿足s[i]==t[i],那麼加上w[i],處理完s和t的所有n位之後的結果即爲這兩個01串的“Wu”值。

樣例解釋:第1行爲n,m,q;第2行爲w[1]...w[n],表示01串第i位的權值爲w[i];之後m行爲multiset裏的m個01串;之後q行爲q次詢問。以樣例1中的第一次詢問爲例:t=00,k=20。與multiset中的每個元素比較(假設第i個元素爲m[i]):

m[1]:01,與00相同的是第一位,所以權值s=w[1]=40;

m[2]:01,與00相同的是第一位,所以權值s=w[1]=40;

m[3]:10,與00相同的是第二位,所以權值s=w[2]=20;

m[4]:11,沒有某一位與00相同,所以權值s=0;

而上述四種比較結果中s<=20的是後兩種,所以輸出2。之後的詢問以此類推。

思路:發現m,q都特別大(1e5),而n只有1-12,即長度最長的01串只有12位,所以可以考慮狀壓來表示每一個01串的狀態,共有2^12種狀態。這樣可以通過枚舉詢問中t和multiset中s轉化爲10進制數後的值來進行預處理,而每一次要處理n位,所以複雜度爲O((2^n)*(2^n)*n),最壞情況大約是2e8。保存一個二維數組sum[4096][100],sum[i][j]表示詢問中輸入的t對應的二進制數i在與multiset中的元素s匹配時結果爲j的元素個數。詳見代碼。

AC代碼:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int w[15],cnt[4100],sum[4100][105];
void solve(){
    int n,m,q;scanf("%d%d%d",&n,&m,&q);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    for(int i=1;i<=m;i++){
        char s[15];scanf("%s",s);
        int x=0;
        for(int j=0;j<n;j++) if(s[j]=='1') x+=1<<(n-j-1);
        cnt[x]++;
    }
    for(int i=0;i<(1<<12);i++){
        for(int j=0;j<(1<<12);j++){
            if(!cnt[j]) continue;
            int x=0;
            for(int k=0;k<12;k++){
                if((i&(1<<k))==(j&(1<<k))) x+=w[n-k];
                if(x>100) break;
            }
            if(x<=100) sum[i][x]+=cnt[j];
        }
    }
    while(q--){
        char s[15];int k;scanf("%s%d",s,&k);
        int x=0;for(int i=0;i<n;i++) if(s[i]=='1') x+=1<<(n-i-1);
        int ans=0;for(int i=0;i<=k;i++) ans+=sum[x][i];
        printf("%d\n",ans);
    }
}
int main(void){
    solve();
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章