hihoCoder [Offer收割]编程练习赛8【数论，dfs，dp，容斥】

题目：http://hihocoder.com/contest/offers8/problems

【我只是简单地mark一下，看详细题解的请戳这里】

A题肯定L和D的最大公因数肯定是循环节，然后直接暴力模拟即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main() {
    int T;
    scanf("%d",&T);
    while(T--) {
        ll L,D,F;
        scanf("%lld%lld%lld",&L,&F,&D);
        ll maxx = L/__gcd(L,D)* D;
        int flag=0;
        for(ll i=0; i<=maxx; i+=D) {
            if(i/L < (i+F)/L) {
                flag=1;
                break;
            }
        }
        if(flag) printf("NO\n");
        else printf("YES\n");
    }
    return 0;
}

---

B：dfs找连通块，基础题。

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
typedef pair<int,int>pii;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N=500+7;
char s[N][N];
int a[N][N],vis[N][N];
vector<pii>q[N*N];
int d[][2]= {{-1,0},{1,0},{0,-1},{0,1}};
int n,m,cnt=0;
void dfs(int x,int y,int id) {
    vis[x][y]=id;
    q[id].push_back(pii(x,y));
    for(int i=0; i<4; i++) {
        int xx=x+d[i][0],yy=y+d[i][1];
        if(xx<0||xx>=n||yy<0||yy>=m||vis[xx][yy]||s[xx][yy]=='0')continue;
        dfs(xx,yy,id);
    }
}
int main() {
    //freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=0; i<n; i++) {
        scanf("%s",s[i]);
    }
    for(int j=0; j<m; j++) {
        for(int i=0; i<n; i++) {
            if(s[i][j]=='1'&&!vis[i][j]) {
                dfs(i,j,++cnt);
            }
        }
    }

    for(int i=1; i<=cnt; i++) {
        int minx=INF,maxx=0,miny=INF,maxy=0;
        for(int j=0; j<q[i].size(); j++) {
            pii& t=q[i][j];
            minx=min(t.fi,minx);
            maxx=max(t.fi,maxx);
            miny=min(miny,t.se);
            maxy=max(maxy,t.se);
            a[t.fi][t.se]=1;
        }
        int R=maxx-minx+1,C=maxy-miny+1;
        for(int r=0; r<R; r++)for(int c=0; c<C; c++)a[r][c]=0;
        for(int j=0; j<q[i].size(); j++) {
            pii& t=q[i][j];
            a[t.fi-minx][t.se-miny]=1;
        }

        printf("%d %d\n",R,C);
        for(int r=0; r<R; r++) {
            for(int c=0; c<C; c++)
                printf("%c",a[r][c]+'0');
            printf("\n");
        }
    }
    return 0;
}

---

C题：简单dp，思路戳上面题解链接。

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
const int N=1e5+9;
void add(int &a,int b){a+=b;if(a>=mod)a-=mod;}

int a[N],dp[20000005],M=10000000;
int main(){
    int T,i,j,k,ca=0,K=0,n,m;
    scanf("%d",&n);
    for(int i=0;i<n;i++)scanf("%d",&a[i]);
    dp[M]=1;
    int s=0,w,sum=1;
    for(int i=0;i<n;i++){
        s+=a[i];
        w=(sum-dp[s+M]+mod)%mod;
        add(dp[s+M],w);
        add(sum,w);
    }
    printf("%d\n",w);
    return 0;
}

---

D题：
忘了在哪里看见一道及其相似的题目了，不过那题是只找正方形。
矩形计数+容斥原理。

（代码直接粘上面题解的了，代码比较清晰，思路戳链接）

#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long LL;

int n, m, k;
int r[10], c[10];

int main()
{
    scanf("%d%d%d", &n, &m, &k);

    LL ans = 0;
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j++)
            ans += (LL)(n + 1 - i) * (m + 1 - j);
    for(int i = 0; i < k; i++)
        scanf("%d%d", r + i, c + i);
    for(int S = 1; S < (1 << k); S++) {
        int lft = m + 1, top = n + 1;
        int down = 0, rgh = 0;
        int cnt = 0;
        for(int i = 0; i < k; i++) if((S >> i) & 1) {
            cnt++;
            lft = min(lft, c[i]);
            top = min(top, r[i]);
            rgh = max(rgh, c[i]);
            down = max(down, r[i]);
        }
        int sign = (cnt & 1) ? -1 : 1;
        ans += (LL)sign * lft * (m + 1 - rgh) * top * (n + 1 - down);
    }

    printf("%lld\n", ans);

    return 0;
}